Commit
·
3b1fc7c
1
Parent(s):
e09cd05
cleaning main
Browse files
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=
|
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():
|