diff --git a/backend/packages/gizmo-agent/gizmo_agent/agent_types/__init__.py b/backend/packages/gizmo-agent/gizmo_agent/agent_types/__init__.py index b47591b3..e9b139f6 100644 --- 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 index 7553206b..a292f843 100644 --- 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(