build: support cohere v2 api

TBXark · Oct 31, 2024 · 62b8147 · 62b8147
1 parent 5ff2ee1
commit 62b8147
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 55 deletions.
diff --git a/dist/buildinfo.json b/dist/buildinfo.json
diff --git a/dist/index.js b/dist/index.js
diff --git a/dist/timestamp b/dist/timestamp
diff --git a/src/agent/cohere.ts b/src/agent/cohere.ts
@@ -1,9 +1,7 @@
 import type { AgentUserConfig } from '../config/env';
 import type { SseChatCompatibleOptions } from './request';
-import type { SSEMessage, SSEParserResult } from './stream';
-import type { ChatAgent, ChatStreamTextHandler, HistoryItem, LLMChatParams } from './types';
+import type { ChatAgent, ChatStreamTextHandler, LLMChatParams } from './types';
 import { requestChatCompletions } from './request';
-import { Stream } from './stream';
 
 export class Cohere implements ChatAgent {
     readonly name = 'cohere';
@@ -17,31 +15,6 @@ export class Cohere implements ChatAgent {
         return ctx.COHERE_CHAT_MODEL;
     };
 
-    private render = (item: HistoryItem): any => {
-        return {
-            role: item.role,
-            message: item.content,
-        };
-    };
-
-    static parser(sse: SSEMessage): SSEParserResult {
-        switch (sse.event) {
-            case 'content-delta':
-                try {
-                    return { data: JSON.parse(sse.data || '') };
-                } catch (e) {
-                    console.error(e, sse.data);
-                    return {};
-                }
-            case 'stream-start':
-                return {};
-            case '[DONE]':
-                return { finish: true };
-            default:
-                return {};
-        }
-    }
-
     readonly request = async (params: LLMChatParams, context: AgentUserConfig, onStream: ChatStreamTextHandler | null): Promise<string> => {
         const { message, prompt, history } = params;
         const url = `${context.COHERE_API_BASE}/chat`;