From 0880fa28634a9f82ef5f241d7abde7b15e257981 Mon Sep 17 00:00:00 2001 From: Vasilije <8619304+Vasilije1990@users.noreply.github.com> Date: Thu, 28 Mar 2024 20:35:40 +0100 Subject: [PATCH] Added anyscale support --- cognee/config.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cognee/config.py b/cognee/config.py index 2523715fd..3d91efb71 100644 --- a/cognee/config.py +++ b/cognee/config.py @@ -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")