Merge pull request #27 from jina-ai/fix-import

fix: dep import in babyagi
jina-ai · Apr 21, 2023 · 7ddd307 · 7ddd307
2 parents 5a56967 + 2e60c98
commit 7ddd307
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lcserve/apps/babyagi/babyagi.py b/lcserve/apps/babyagi/babyagi.py
@@ -2,7 +2,6 @@
 from collections import deque
 from typing import Any, Dict, List, Literal, Optional, Union
 
-import faiss
 from fastapi import WebSocket
 from langchain import LLMChain, OpenAI, PromptTemplate, SerpAPIWrapper
 from langchain.agents import AgentExecutor, Tool, ZeroShotAgent
@@ -17,6 +16,8 @@
 
 
 def get_vectorstore():
+    import faiss
+
     # Define your embedding model
     embeddings_model = OpenAIEmbeddings()