wip

Author: harel-tussi

ai21/clients/common/agents/agents.py

@@ -89,7 +89,6 @@ def create(
         tool_resources: Dict[str, Any] | NotGiven = NOT_GIVEN,
         requirements: List[AgentRequirement] | NotGiven = NOT_GIVEN,
         budget: BudgetLevel | NotGiven = NOT_GIVEN,
-        agent_type: AgentType | NotGiven = NOT_GIVEN,
         **kwargs,
     ) -> Agent:
         pass

examples/studio/agents/agent_crud.py

@@ -16,7 +16,6 @@ def main():
     )
 
     print(f"Created agent: {agent.name} (ID: {agent.id})")
-    print(f"Agent type: {agent.agent_type}")
     print(f"Budget: {agent.budget}")
 
     agent_id = agent.id

tests/integration_tests/clients/studio/test_agents.py

@@ -1,7 +1,7 @@
 import pytest
 
 from ai21 import AsyncAI21Client
-from ai21.models.agents import BudgetLevel, AgentType
+from ai21.models.agents import BudgetLevel
 
 
 @pytest.mark.asyncio
@@ -21,7 +21,6 @@ async def test_agents_create_and_delete():
     assert agent.name == "Test Agent for Integration"
     assert agent.description == "This is a test agent created by integration tests"
     assert agent.budget == BudgetLevel.LOW
-    assert agent.agent_type == AgentType.DEFAULT
     assert agent.object == "agent"
 
     # Clean up - delete the agent