Python: metadata for agent.run

[hoangtnm]

I need my agent to call tools that require access_token for authentication. My initial approach was to add access_token as a parameter to the tool function (e.g., get_weather). However, agent.run(...) only accepts a single text string as input, so I can't pass additional arguments (e.g., access_token).

from typing import Annotated
from pydantic import Field

def get_weather(
    location: Annotated[str, Field(description="The location to get the weather for.")],
    access_token: Annotated[str, Field(description="Access token for authentication.")],
) -> str:
    """Get the weather for a given location."""
    data = auth(access_token)  # Authenticate using the token
    return f"The weather in {data['location']} is cloudy with a high of 15°C."

# Problem: How do I pass access_token when calling the agent?
result = await agent.run("What is the weather like in Amsterdam?")

I am just wondering if it is possible to provide authentication credentials or metadata to tools when the agent's run method only accepts a text query?

I expect something like this:

def get_weather(
    location: Annotated[str, Field(description="The location to get the weather for.")],
    access_token: Annotated[str, Field(description="Access token for authentication.")],
) -> str:
    """Get the weather for a given location."""
    data = auth(access_token)  # Authenticate using the token
    return f"The weather in {data['location']} is cloudy with a high of 15°C."

# Problem: How do I pass access_token when calling the agent?
result = await agent.run("What is the weather like in Amsterdam?", access_token="abc")

[markwallace-microsoft]

@hoangtnm You should be able to use function middleware for this but we don't have a sample for this specific use case. I'll use this issue to track getting the sample created.

@dmytrostruk and @rogerbarreto can we create samples for this scenario to validate this will work. This is the sample we had in SK: https://github.com/microsoft/semantic-kernel/blob/43da976c9b42548088e51e326b7df90bdc57a056/dotnet/samples/Concepts/Plugins/TransformPlugin.cs

[rogerbarreto]

@hoangtnm In .NET the most straightforward way to achieve this would be using the IServiceProvider parameter in your functions, this automatically injects that to your function awareness, where you would have your TokenService registered and the function could access it via services.GetRequiredService<TokenService>() within you function code.

For C# Please have a look on our Agent_Step15_Plugins sample using this capability.