feat: Agents in CLI (#776)

Co-authored-by: Konrad Czarnota <[email protected]>
Co-authored-by: GlockPL <[email protected]>
Co-authored-by: akotyla <[email protected]>
Co-authored-by: jakubduda-dsai <[email protected]>
Co-authored-by: ds-sebastianchwilczynski <[email protected]>
Co-authored-by: dazy-ds <[email protected]>
Co-authored-by: pocucan-ds <[email protected]>

Author: 7 people

examples/agents/cli_agent.py

@@ -0,0 +1,42 @@
+"""Simple test agent for CLI testing.
+
+run:
+
+uv run ragbits agents run examples.agents.test_cli_agent:test_agent
+
+"""
+
+from ragbits.agents import Agent
+from ragbits.core.llms import LiteLLM
+
+
+def get_current_weather(location: str) -> str:
+    """
+    Get the current weather in a specific location.
+
+    Args:
+        location: The location to get weather for
+
+    Returns:
+        Weather information for the location
+    """
+    # Mock weather data
+    weather_data = {
+        "London": "Rainy, 15°C",
+        "Paris": "Sunny, 22°C",
+        "New York": "Cloudy, 18°C",
+        "Tokyo": "Sunny, 25°C",
+    }
+
+    return weather_data.get(location, f"Weather data not available for {location}")
+
+
+# Create a simple test agent
+llm = LiteLLM(model_name="gpt-3.5-turbo")
+
+test_agent = Agent(
+    llm=llm,
+    prompt="You are a helpful assistant that can help with weather information and general questions.",
+    tools=[get_current_weather],
+    keep_history=True,
+)

packages/ragbits-agents/CHANGELOG.md

@@ -1,6 +1,7 @@
 # CHANGELOG
 
 ## Unreleased
+- Add Chat with agents in CLI
 - Add tool_choice parameter to agent interface (#738)
 - Add PydanticAI agents support (#755)
 - Add unique ID to each agent instance for better tracking and identification (#790)

packages/ragbits-agents/pyproject.toml

@@ -45,6 +45,9 @@ mcp = [
 openai = [
     "openai>=1.91.0,<2.0.0",
 ]
+cli = [
+    "textual>=0.85.2,<1.0.0",
+]
 
 [project.urls]
 "Homepage" = "https://github.com/deepsense-ai/ragbits"