gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Added anyscale support

Browse source
This commit is contained in:
Vasilije 2024-03-28 20:35:40 +01:00
parent 277acf081b
commit 0880fa2863
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
cognee/config.py
View file

@ -43,7 +43,7 @@ class Config:
graph_filename = os.getenv("GRAPH_NAME", "cognee_graph.pkl")
# Model parameters
llm_provider: str = "custom" #openai, or custom or ollama
llm_provider: str = "openai" #openai, or custom or ollama
custom_model: str = "mistralai/Mixtral-8x7B-Instruct-v0.1"
custom_endpoint: str = "https://api.endpoints.anyscale.com/v1" # pass claude endpoint
custom_key: Optional[str] = os.getenv("ANYSCALE_API_KEY")