Refactor agent retrieval methods to use updated API calls and update requirements for new package version

Author: howieleung

src/api/main.py

@@ -54,7 +54,7 @@ async def lifespan(app: fastapi.FastAPI):
                     try: 
                         agent_name = agent_id.split(":")[0]
                         agent_version = agent_id.split(":")[1]
-                        agent_version_obj = await ai_project.agents.retrieve_version(agent_name, agent_version)
+                        agent_version_obj = await ai_project.agents.get_version(agent_name, agent_version)
                         logger.info("Agent already exists, skipping creation")
                         logger.info(f"Fetched agent, agent ID: {agent_version_obj.id}")
                     except Exception as e:

src/gunicorn.conf.py

@@ -295,7 +295,7 @@ async def initialize_resources():
                 try:
                     agent_name = agentID.split(":")[0]
                     agent_version = agentID.split(":")[1]
-                    agent_obj = await ai_project.agents.retrieve_version(agent_name, agent_version)
+                    agent_obj = await ai_project.agents.get_version(agent_name, agent_version)
                     logger.info(f"Found agent by ID: {agent_obj.id}")
                 except Exception as e:
                     logger.warning(
@@ -309,7 +309,7 @@ async def initialize_resources():
                 try:
                     agent_name = os.environ["AZURE_AI_AGENT_NAME"]
                     logger.info(f"Retrieving agent by name: {agent_name}")
-                    agents = await ai_project.agents.retrieve(agent_name)
+                    agents = await ai_project.agents.get(agent_name)
                     agent_obj = agents.versions.latest
                     logger.info(f"Agent with agent id, {agent_obj.id} retrieved.")
                 except Exception as e:

src/packages/azure_ai_projects-2.0.0a1105001-py3-none-any.whl

@@ -4,7 +4,7 @@ gunicorn==23.0.0
 azure-identity==1.19.0
 aiohttp==3.13.1
 openai
-./packages/azure_ai_projects-2.0.0a20251103001-py3-none-any.whl
+./packages/azure_ai_projects-2.0.0a20251117001-py3-none-any.whl
 azure-core==1.36.0  # other versions might not compatible
 azure-core-tracing-opentelemetry==1.0.0b12
 azure-monitor-opentelemetry-exporter==1.0.0b44

src/packages/azure_ai_projects-2.0.0a20251117001-py3-none-any.whl

@@ -36,7 +36,7 @@ def test_evaluation():
 
         openai_client = project_client.get_openai_client()
 
-        agent = project_client.agents.retrieve_version(
+        agent = project_client.agents.get_version(
             agent_name=agent_name, agent_version=agent_version
         )
         print(f"Agent retrieved (id: {agent.id}, name: {agent.name}, version: {agent.version})")

src/requirements.txt

@@ -40,13 +40,13 @@ def retrieve_agent(project_client: AIProjectClient) -> AgentVersionObject:
 
     print(f"Retrieving agent: {agent_name}")
     if agent_version:
-        agent = project_client.agents.retrieve_version(
+        agent = project_client.agents.get_version(
             agent_name=agent_name, agent_version=agent_version
         )
         print(f"Retrieved agent: {agent_name}, version: {agent_version}")
         return agent
     else:
-        agent_versions = project_client.agents.retrieve(agent_name=agent_name)
+        agent_versions = project_client.agents.get(agent_name=agent_name)
         agent_version = agent_versions.versions.latest
         print(f"Retrieved agent: {agent_name}, with latest version: {agent_version}")
         return agent_versions.versions.latest