[feat] Add current_user_id and current_session_id (#3152)

## Summary

Keep track of the current user ID and current session ID to use in tools

## Type of change

- [ ] Bug fix
- [x] New feature
- [ ] Breaking change
- [ ] Improvement
- [ ] Model update
- [ ] Other:

---

## Checklist

- [ ] Code complies with style guidelines
- [ ] Ran format/validation scripts (`./scripts/format.sh` and
`./scripts/validate.sh`)
- [ ] Self-review completed
- [ ] Documentation updated (comments, docstrings)
- [ ] Examples and guides: Relevant cookbook examples have been included
or updated (if applicable)
- [ ] Tested in clean environment
- [ ] Tests added/updated (if applicable)

---

## Additional Notes

Add any important context (deployment instructions, screenshots,
security considerations, etc.)

---------

Co-authored-by: Kaustubh <shuklakaustubh84@gmail.com>

Author: dirkbrnd

cookbook/agent_concepts/state/session_state_user_id.py

@@ -0,0 +1,123 @@
+"""
+This example demonstrates how to maintain state for each user in a multi-user environment.
+
+The shopping list is stored in a dictionary, organized by user ID and session ID.
+
+Agno automatically creates the "current_user_id" and "current_session_id" variables in the session state.
+
+You can access these variables in your functions using the `agent.session_state` dictionary.
+"""
+
+import json
+
+from agno.agent import Agent
+from agno.models.openai import OpenAIChat
+from agno.utils.log import log_info
+
+# In-memory database to store user shopping lists
+# Organized by user ID and session ID
+shopping_list = {}
+
+
+def add_item(agent: Agent, item: str) -> str:
+    """Add an item to the current user's shopping list."""
+    current_user_id = agent.session_state["current_user_id"]
+    current_session_id = agent.session_state["current_session_id"]
+    shopping_list.setdefault(current_user_id, {}).setdefault(
+        current_session_id, []
+    ).append(item)
+    return f"Item {item} added to the shopping list"
+
+
+def remove_item(agent: Agent, item: str) -> str:
+    """Remove an item from the current user's shopping list."""
+    current_user_id = agent.session_state["current_user_id"]
+    current_session_id = agent.session_state["current_session_id"]
+
+    if (
+        current_user_id not in shopping_list
+        or current_session_id not in shopping_list[current_user_id]
+    ):
+        return f"No shopping list found for user {current_user_id} and session {current_session_id}"
+
+    if item not in shopping_list[current_user_id][current_session_id]:
+        return f"Item '{item}' not found in the shopping list for user {current_user_id} and session {current_session_id}"
+
+    shopping_list[current_user_id][current_session_id].remove(item)
+    return f"Item {item} removed from the shopping list"
+
+
+def get_shopping_list(agent: Agent) -> str:
+    """Get the current user's shopping list."""
+    current_user_id = agent.session_state["current_user_id"]
+    current_session_id = agent.session_state["current_session_id"]
+    return f"Shopping list for user {current_user_id} and session {current_session_id}: \n{json.dumps(shopping_list[current_user_id][current_session_id], indent=2)}"
+
+
+# Create an Agent that maintains state
+agent = Agent(
+    model=OpenAIChat(id="gpt-4o-mini"),
+    tools=[add_item, remove_item, get_shopping_list],
+    # Reference the in-memory database
+    instructions=[
+        "Current User ID: {current_user_id}",
+        "Current Session ID: {current_session_id}",
+    ],
+    # Important: Add the state in the instructions
+    add_state_in_messages=True,
+    markdown=True,
+)
+
+user_id_1 = "john_doe"
+user_id_2 = "mark_smith"
+user_id_3 = "carmen_sandiago"
+
+# Example usage
+agent.print_response(
+    "Add milk, eggs, and bread to the shopping list",
+    stream=True,
+    user_id=user_id_1,
+    session_id="user_1_session_1",
+)
+agent.print_response(
+    "Add tacos to the shopping list",
+    stream=True,
+    user_id=user_id_2,
+    session_id="user_2_session_1",
+)
+agent.print_response(
+    "Add apples and grapesto the shopping list",
+    stream=True,
+    user_id=user_id_3,
+    session_id="user_3_session_1",
+)
+agent.print_response(
+    "Remove milk from the shopping list",
+    stream=True,
+    user_id=user_id_1,
+    session_id="user_1_session_1",
+)
+agent.print_response(
+    "Add minced beef to the shopping list",
+    stream=True,
+    user_id=user_id_2,
+    session_id="user_2_session_1",
+)
+
+# What is on Mark Smith's shopping list?
+agent.print_response(
+    "What is on Mark Smith's shopping list?",
+    stream=True,
+    user_id=user_id_2,
+    session_id="user_2_session_1",
+)
+
+# New session, so new shopping list
+agent.print_response(
+    "Add chicken and soup to my list.",
+    stream=True,
+    user_id=user_id_2,
+    session_id="user_3_session_2",
+)
+
+print(f"Final shopping lists: \n{json.dumps(shopping_list, indent=2)}")

cookbook/observability/langfuse_via_openlit.py

@@ -2,47 +2,54 @@
 This example shows how to use langfuse via OpenLIT to trace model calls.
 
 1. Install dependencies: pip install openai langfuse openlit opentelemetry-sdk opentelemetry-exporter-otlp
-2. Set your Langfuse API key as an environment variables: 
+2. Set your Langfuse API key as an environment variables:
   - export LANGFUSE_PUBLIC_KEY=<your-key>
   - export LANGFUSE_SECRET_KEY=<your-key>
 """
 
 import base64
 import os
+
 from agno.agent import Agent
 from agno.models.openai import OpenAIChat
 from agno.tools.duckduckgo import DuckDuckGoTools
 
-LANGFUSE_AUTH = base64.b64encode(f"{os.getenv('LANGFUSE_PUBLIC_KEY')}:{os.getenv('LANGFUSE_SECRET_KEY')}".encode()).decode()
+LANGFUSE_AUTH = base64.b64encode(
+    f"{os.getenv('LANGFUSE_PUBLIC_KEY')}:{os.getenv('LANGFUSE_SECRET_KEY')}".encode()
+).decode()
 
-os.environ["OTEL_EXPORTER_OTLP_ENDPOINT"]="https://us.cloud.langfuse.com/api/public/otel" # 🇺🇸 US data region
+os.environ["OTEL_EXPORTER_OTLP_ENDPOINT"] = (
+    "https://us.cloud.langfuse.com/api/public/otel"  # 🇺🇸 US data region
+)
 # os.environ["OTEL_EXPORTER_OTLP_ENDPOINT"]="https://cloud.langfuse.com/api/public/otel" # 🇪🇺 EU data region
 # os.environ["OTEL_EXPORTER_OTLP_ENDPOINT"]="http://localhost:3000/api/public/otel" # 🏠 Local deployment (>= v3.22.0)
- 
-os.environ["OTEL_EXPORTER_OTLP_HEADERS"]=f"Authorization=Basic {LANGFUSE_AUTH}"
 
-from opentelemetry.sdk.trace import TracerProvider
+os.environ["OTEL_EXPORTER_OTLP_HEADERS"] = f"Authorization=Basic {LANGFUSE_AUTH}"
+
 from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter
+from opentelemetry.sdk.trace import TracerProvider
 from opentelemetry.sdk.trace.export import SimpleSpanProcessor
- 
+
 trace_provider = TracerProvider()
 trace_provider.add_span_processor(SimpleSpanProcessor(OTLPSpanExporter()))
- 
+
 # Sets the global default tracer provider
 from opentelemetry import trace
+
 trace.set_tracer_provider(trace_provider)
- 
+
 # Creates a tracer from the global tracer provider
 tracer = trace.get_tracer(__name__)
 
 import openlit
+
 # Initialize OpenLIT instrumentation. The disable_batch flag is set to true to process traces immediately.
 openlit.init(tracer=tracer, disable_batch=True)
 
 agent = Agent(
-    model=OpenAIChat(id="gpt-4o-mini"), 
+    model=OpenAIChat(id="gpt-4o-mini"),
     tools=[DuckDuckGoTools()],
-    markdown=True, 
+    markdown=True,
     debug_mode=True,
 )
 

libs/agno/agno/agent/agent.py

@@ -1089,6 +1089,8 @@ def run(
 
         session_id = cast(str, session_id)
 
+        self._initialize_session_state(user_id=user_id, session_id=session_id)
+
         log_debug(f"Session ID: {session_id}", center=True)
 
         last_exception = None
@@ -1698,6 +1700,8 @@ async def arun(
 
         session_id = cast(str, session_id)
 
+        self._initialize_session_state(user_id=user_id, session_id=session_id)
+
         log_debug(f"Session ID: {session_id}", center=True)
 
         last_exception = None
@@ -1849,6 +1853,11 @@ def create_run_response(
             rr.created_at = created_at
         return rr
 
+    def _initialize_session_state(self, user_id: Optional[str] = None, session_id: Optional[str] = None) -> None:
+        self.session_state = self.session_state or {}
+        self.session_state["current_user_id"] = user_id
+        self.session_state["current_session_id"] = session_id
+
     def _make_memories_and_summaries(
         self,
         run_messages: RunMessages,

libs/agno/agno/team/team.py

@@ -567,6 +567,8 @@ def run(
 
         session_id = cast(str, session_id)
 
+        self._initialize_session_state(user_id=user_id, session_id=session_id)
+
         log_debug(f"Session ID: {session_id}", center=True)
 
         self.initialize_team(session_id=session_id)
@@ -1313,6 +1315,8 @@ async def arun(
 
         session_id = cast(str, session_id)
 
+        self._initialize_session_state(user_id=user_id, session_id=session_id)
+
         log_debug(f"Session ID: {session_id}", center=True)
 
         self.initialize_team(session_id=session_id)
@@ -1990,6 +1994,11 @@ async def _arun_stream(
 
         log_debug(f"Team Run End: {self.run_id}", center=True, symbol="*")
 
+    def _initialize_session_state(self, user_id: Optional[str] = None, session_id: Optional[str] = None) -> None:
+        self.session_state = self.session_state or {}
+        self.session_state["current_user_id"] = user_id
+        self.session_state["current_session_id"] = session_id
+
     def _make_memories_and_summaries(
         self, run_messages: RunMessages, session_id: str, user_id: Optional[str] = None
     ) -> None: