Skip to content

Commit 61baa10

Browse files
committed
Clearer client
1 parent 23d9e92 commit 61baa10

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

examples/agentframework_basic.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -14,15 +14,14 @@
1414
async_credential = None
1515
if API_HOST == "azure":
1616
async_credential = DefaultAzureCredential()
17-
client = OpenAIChatClient(
18-
base_url=os.environ.get("AZURE_OPENAI_ENDPOINT") + "/openai/v1/", api_key=get_bearer_token_provider(async_credential, "https://cognitiveservices.azure.com/.default"), model_id=os.environ.get("AZURE_OPENAI_CHAT_DEPLOYMENT")
19-
)
17+
token_provider = get_bearer_token_provider(async_credential, "https://cognitiveservices.azure.com/.default")
18+
client = OpenAIChatClient(base_url=os.environ["AZURE_OPENAI_ENDPOINT"] + "/openai/v1/", api_key=token_provider, model_id=os.environ["AZURE_OPENAI_CHAT_DEPLOYMENT"])
2019
elif API_HOST == "github":
2120
client = OpenAIChatClient(base_url="https://models.github.ai/inference", api_key=os.environ["GITHUB_TOKEN"], model_id=os.getenv("GITHUB_MODEL", "openai/gpt-4o"))
2221
elif API_HOST == "ollama":
2322
client = OpenAIChatClient(base_url=os.environ.get("OLLAMA_ENDPOINT", "http://localhost:11434/v1"), api_key="none", model_id=os.environ.get("OLLAMA_MODEL", "llama3.1:latest"))
2423
else:
25-
client = OpenAIChatClient(api_key=os.environ.get("OPENAI_API_KEY"), model_id=os.environ.get("OPENAI_MODEL", "gpt-4o"))
24+
client = OpenAIChatClient(api_key=os.environ["OPENAI_API_KEY"], model_id=os.environ.get("OPENAI_MODEL", "gpt-4o"))
2625

2726
agent = ChatAgent(chat_client=client, instructions="You're an informational agent. Answer questions cheerfully.")
2827

0 commit comments

Comments
 (0)