Grazie LLM fixed

JetBrains-Research · Jul 10, 2024 · 44a11aa · 44a11aa
1 parent e81db58
commit 44a11aa
Show file tree

Hide file tree

Showing 4 changed files with 594 additions and 682 deletions.
diff --git a/code_editing/configs/agents/llm_config.py b/code_editing/configs/agents/llm_config.py
@@ -28,7 +28,7 @@ class LLMProfileConfig:
     grazie_jwt_token: str = os.environ.get("GRAZIE_JWT_TOKEN", MISSING)
     auth_type: str = AuthType.USER
     client_url: str = GrazieApiGatewayUrls.STAGING
-    profile: LLMProfileConfig = field(default_factory=LLMProfileConfig)
+    profile: str = "openai-chat-gpt"
 
 
 cs = ConfigStore.instance()
@@ -42,6 +42,6 @@ def create_agent_method(llm_config: ChatLLMConfig):
     if target == "langchain_openai.ChatOpenAI":
         return "langchain.agents.create_openai_tools_agent"
     elif target == "grazie_langchain_utils.language_models.grazie.ChatGrazie":
-        return "grazie_langchain_utils.agents.create_grazie_tools_agent"
+        return "grazie_langchain_utils.agents.create_tool_calling_agent"
     else:
         raise ValueError(f"Unknown target: {target}")
diff --git a/code_editing/scripts/conf/llm/claude-3-sonnet.yaml b/code_editing/scripts/conf/llm/claude-3-sonnet.yaml
@@ -0,0 +1,5 @@
+defaults:
+  - grazie
+  - _self_
+
+profile: anthropic-claude-3-sonnet