Phoenix21 commited on
Commit
d9f3ab5
·
verified ·
1 Parent(s): 0dbe352

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +1 -1
pipeline.py CHANGED
@@ -12,7 +12,7 @@ from langchain.embeddings import HuggingFaceEmbeddings
12
  from langchain.vectorstores import FAISS
13
  from langchain.chains import RetrievalQA
14
 
15
- from smolagents import DuckDuckGoSearchTool, ManagedAgent
16
  from pydantic import BaseModel, Field, ValidationError, validator
17
  from mistralai import Mistral
18
 
 
12
  from langchain.vectorstores import FAISS
13
  from langchain.chains import RetrievalQA
14
 
15
+ from smolagents import DuckDuckGoSearchTool, ManagedAgent, LiteLLMModel
16
  from pydantic import BaseModel, Field, ValidationError, validator
17
  from mistralai import Mistral
18