Tonic commited on
Commit
d24a7c2
1 Parent(s): 66854bf

add chroma cllection name

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -130,7 +130,7 @@ class ChromaManager:
130
  intention_client = OpenAI(api_key=yi_token, base_url=API_BASE)
131
  embedding_generator = EmbeddingGenerator(model_name=model_name, token=hf_token, intention_client=intention_client)
132
  embedding_function = MyEmbeddingFunction(embedding_generator=embedding_generator)
133
- chroma_manager = ChromaManager(embedding_function=embedding_function)
134
 
135
  def respond(
136
  message,
 
130
  intention_client = OpenAI(api_key=yi_token, base_url=API_BASE)
131
  embedding_generator = EmbeddingGenerator(model_name=model_name, token=hf_token, intention_client=intention_client)
132
  embedding_function = MyEmbeddingFunction(embedding_generator=embedding_generator)
133
+ chroma_manager = ChromaManager(collection_name="Tonic-instruct" , embedding_function=embedding_function)
134
 
135
  def respond(
136
  message,