Disable claude

langchain-ai · Dec 14, 2023 · 34e6ae7 · 34e6ae7
1 parent a3db922
commit 34e6ae7
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/backend/packages/gizmo-agent/gizmo_agent/agent_types/__init__.py b/backend/packages/gizmo-agent/gizmo_agent/agent_types/__init__.py
@@ -8,8 +8,8 @@ class GizmoAgentType(str, Enum):
     GPT_35_TURBO = "GPT 3.5 Turbo"
     # GPT_4 = "GPT 4"
     # AZURE_OPENAI = "GPT 4 (Azure OpenAI)"
-    CLAUDE2 = "Claude 2"
-    BEDROCK_CLAUDE2 = "Claude 2 (Amazon Bedrock)"
+    # CLAUDE2 = "Claude 2"
+    # BEDROCK_CLAUDE2 = "Claude 2 (Amazon Bedrock)"
 
 
 __all__ = [

diff --git a/backend/packages/gizmo-agent/gizmo_agent/main.py b/backend/packages/gizmo-agent/gizmo_agent/main.py
@@ -57,10 +57,10 @@ def __init__(
         #     _agent = get_openai_function_agent(_tools, system_message, gpt_4=True)
         # elif agent == GizmoAgentType.AZURE_OPENAI:
         #     _agent = get_openai_function_agent(_tools, system_message, azure=True)
-        elif agent == GizmoAgentType.CLAUDE2:
-            _agent = get_xml_agent(_tools, system_message)
-        elif agent == GizmoAgentType.BEDROCK_CLAUDE2:
-            _agent = get_xml_agent(_tools, system_message, bedrock=True)
+        # elif agent == GizmoAgentType.CLAUDE2:
+        #     _agent = get_xml_agent(_tools, system_message)
+        # elif agent == GizmoAgentType.BEDROCK_CLAUDE2:
+        #     _agent = get_xml_agent(_tools, system_message, bedrock=True)
         else:
             raise ValueError("Unexpected agent type")
         agent_executor = get_agent_executor(