Tonic commited on
Commit
17d91f4
β€’
1 Parent(s): 1d30978

added langchain app for simplicity

Browse files
Files changed (1) hide show
  1. langchainapp.py +1 -1
langchainapp.py CHANGED
@@ -184,7 +184,7 @@ def answer_query(message: str, chat_history: List[Tuple[str, str]]):
184
 
185
  # Initialize clients
186
  intention_client = OpenAI(api_key=yi_token, base_url=API_BASE)
187
- embedding_function = MyEmbeddingFunction(model_name=model_name, intention_client=intention_client)
188
  chroma_db = initialize_chroma(collection_name="Tonic-instruct", embedding_function=embedding_function)
189
 
190
  def upload_documents(files):
 
184
 
185
  # Initialize clients
186
  intention_client = OpenAI(api_key=yi_token, base_url=API_BASE)
187
+ embedding_function = MyEmbeddingFunction(model_name=model_name, token=hf_token, intention_client=intention_client)
188
  chroma_db = initialize_chroma(collection_name="Tonic-instruct", embedding_function=embedding_function)
189
 
190
  def upload_documents(files):