Spaces:

Tonic
/

YiJina

Build error

Tonic commited on Jul 13

Commit

399e250

•

1 Parent(s): 51fa852

fix chroma in memory instantiation

Files changed (1) hide show

app.py CHANGED Viewed

@@ -157,7 +157,7 @@ def query_chroma(query_text: str, embedding_function: MyEmbeddingFunction):
 intention_client = OpenAI(api_key=yi_token, base_url=API_BASE)
 embedding_generator = EmbeddingGenerator(model_name=model_name, token=hf_token, intention_client=intention_client)
 embedding_function = MyEmbeddingFunction(embedding_generator=embedding_generator)
-chroma_client, chroma_collection = initialize_chroma(collection_name="Tonic-instruct", embedding_function=embedding_function)
 def respond(
     message,

 intention_client = OpenAI(api_key=yi_token, base_url=API_BASE)
 embedding_generator = EmbeddingGenerator(model_name=model_name, token=hf_token, intention_client=intention_client)
 embedding_function = MyEmbeddingFunction(embedding_generator=embedding_generator)
+chroma_db = initialize_chroma(collection_name="Tonic-instruct", embedding_function=embedding_function)
 def respond(
     message,