diff --git a/mem0-ts/src/oss/src/llms/together.ts b/mem0-ts/src/oss/src/llms/together.ts index b76418c86..078295445 100644 --- a/mem0-ts/src/oss/src/llms/together.ts +++ b/mem0-ts/src/oss/src/llms/together.ts @@ -5,20 +5,17 @@ import { LLMResponse } from "./base"; export class TogetherLLM extends OpenAILLM { constructor(config: LLMConfig) { const apiKey = config.apiKey || process.env.TOGETHER_API_KEY; - if (!apiKey) { throw new Error("Together API key is required"); } - super({ ...config, apiKey, baseURL: config.baseURL || + process.env.TOGETHER_API_BASE || "https://api.together.xyz/v1", - model: - config.model || - "mistralai/Mixtral-8x7B-Instruct-v0.1", + model: config.model || "mistralai/Mixtral-8x7B-Instruct-v0.1", }); } @@ -28,37 +25,19 @@ export class TogetherLLM extends OpenAILLM { tools?: any[], ): Promise { try { - return await super.generateResponse( - messages, - responseFormat, - tools, - ); + return await super.generateResponse(messages, responseFormat, tools); } catch (err) { - const message = - err instanceof Error - ? err.message - : String(err); - - throw new Error( - `Together LLM failed: ${message}`, - ); + const message = err instanceof Error ? err.message : String(err); + throw new Error(`Together LLM failed: ${message}`); } } - async generateChat( - messages: Message[], - ): Promise { + async generateChat(messages: Message[]): Promise { try { return await super.generateChat(messages); } catch (err) { - const message = - err instanceof Error - ? err.message - : String(err); - - throw new Error( - `Together LLM failed: ${message}`, - ); + const message = err instanceof Error ? err.message : String(err); + throw new Error(`Together LLM failed: ${message}`); } } -} \ No newline at end of file +} diff --git a/mem0-ts/src/oss/src/utils/factory.ts b/mem0-ts/src/oss/src/utils/factory.ts index 018acb9e2..8ac82d159 100644 --- a/mem0-ts/src/oss/src/utils/factory.ts +++ b/mem0-ts/src/oss/src/utils/factory.ts @@ -1,4 +1,3 @@ - import { OpenAIEmbedder } from "../embeddings/openai"; import { OllamaEmbedder } from "../embeddings/ollama"; import { LMStudioEmbedder } from "../embeddings/lmstudio";