chore: update template

Azure-Samples · Jul 2, 2024 · 7f910c1 · 7f910c1
1 parent 8813de2
commit 7f910c1
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/scripts/setup-template.sh b/scripts/setup-template.sh
@@ -129,7 +129,8 @@ if [ "$template_name" == "qdrant" ]; then
   echo -e "import fp from 'fastify-plugin';
 import { type BaseChatModel } from '@langchain/core/language_models/chat_models';
 import { type VectorStore } from '@langchain/core/vectorstores';
-import { type Message, MessageBuilder, type ChatResponse, type ChatResponseChunk } from '../lib/index.js';
+import { AIChatMessage, AIChatCompletionDelta, AIChatCompletion } from '@microsoft/ai-chat-protocol';
+import { MessageBuilder } from '../lib/message-builder.js';
 import { type AppConfig } from './config.js';
 
 export class ChatService {
@@ -208,7 +209,8 @@ elif [ "$template_name" == "aisearch" ]; then
   echo -e "import fp from 'fastify-plugin';
 import { type BaseChatModel } from '@langchain/core/language_models/chat_models';
 import { type VectorStore } from '@langchain/core/vectorstores';
-import { type Message, MessageBuilder, type ChatResponse, type ChatResponseChunk } from '../lib/index.js';
+import { AIChatMessage, AIChatCompletionDelta, AIChatCompletion } from '@microsoft/ai-chat-protocol';
+import { MessageBuilder } from '../lib/message-builder.js';
 import { type AppConfig } from './config.js';
 
 export class ChatService {