fix: langchain service typings

Azure-Samples · Sep 1, 2023 · b7548d9 · b7548d9
1 parent f23b342
commit b7548d9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/packages/api/src/plugins/langchain.ts b/packages/api/src/plugins/langchain.ts
@@ -4,8 +4,8 @@ import { ChatOpenAI } from 'langchain/chat_models/openai';
 import { OpenAIEmbeddings } from 'langchain/embeddings/openai';
 
 export type LangchainService = {
-  getChat(): Promise<ChatOpenAI>;
-  getEmbeddings(): Promise<OpenAIEmbeddings>;
+  getChat(options?: Partial<OpenAIChatInput>): Promise<ChatOpenAI>;
+  getEmbeddings(options?: Partial<OpenAIChatInput>): Promise<OpenAIEmbeddings>;
 };
 
 export default fp(