pbrenotvinciguerra commited on
Commit
3b1fc7c
·
1 Parent(s): e09cd05

cleaning main

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -14,7 +14,7 @@ app = FastAPI()
14
  llm = Anyscale(model="mistralai/Mistral-7B-Instruct-v0.1", api_key=os.getenv("ANYSCALE_API_KEY"))
15
  # Define the embedding model used to embed the query.
16
  # query_embed_model = HuggingFaceEmbedding(model_name="BAAI/bge-base-en-v1.5")
17
- embed_model = VoyageEmbedding(model_name=model_name, voyage_api_key=os.getenv("VOYAGE_API_KEY"))
18
  service_context = ServiceContext.from_defaults(llm=llm, embed_model=embed_model)
19
 
20
  if "index" in os.list():
 
14
  llm = Anyscale(model="mistralai/Mistral-7B-Instruct-v0.1", api_key=os.getenv("ANYSCALE_API_KEY"))
15
  # Define the embedding model used to embed the query.
16
  # query_embed_model = HuggingFaceEmbedding(model_name="BAAI/bge-base-en-v1.5")
17
+ embed_model = VoyageEmbedding(model_name="voyage-01", voyage_api_key=os.getenv("VOYAGE_API_KEY"))
18
  service_context = ServiceContext.from_defaults(llm=llm, embed_model=embed_model)
19
 
20
  if "index" in os.list():