Merge pull request #47 from Azure-Samples/howie/get-agent

Fix get agent using env variable

Author: howieleung

src/api/main.py

@@ -84,7 +84,7 @@ async def lifespan(app: fastapi.FastAPI):
             else:
                 configure_azure_monitor(connection_string=application_insights_connection_string)
 
-        if os.environ.get("AZURE_AI_AGENT_ID") is not None:
+        if os.environ.get("AZURE_AI_AGENT_ID"):
             try: 
                 agent = await ai_client.agents.get_agent(os.environ["AZURE_AI_AGENT_ID"])
                 logger.info("Agent already exists, skipping creation")

src/gunicorn.conf.py

@@ -116,7 +116,7 @@ async def initialize_resources():
         )
 
         # If the environment already has AZURE_AI_AGENT_ID, try fetching that agent
-        if os.environ.get("AZURE_AI_AGENT_ID") is not None:
+        if os.environ.get("AZURE_AI_AGENT_ID"):
             try: 
                 agent = await ai_client.agents.get_agent(os.environ["AZURE_AI_AGENT_ID"])
                 logger.info(f"Found agent by ID: {agent.id}")