simplify docker with automatic framework detection

Author: blairhudson

README.md

@@ -57,17 +57,18 @@ Here’s how to get started with a basic `PydanticAI` agent:
 ```python
 from fastapi import FastAPI
 from fastapi_agents import FastAPIAgents
-from fastapi_agents.pydantic_ai import PydanticAIAgent
 from pydantic_ai import Agent
 
+# Create a FastAPI application and FastAPIAgents instance.
 app = FastAPI()
 agents = FastAPIAgents(path_prefix="/agents")
 
-# Initialize and register the agent
+# Initialize and register the agent. 
+# The framework is automatically detected.
 agent = Agent("openai:gpt-4o-mini")
-agents.register("pydanticai", PydanticAIAgent(agent))
+agents.register("pydanticai", agent)
 
-# Include the router
+# Include the router in your FastAPI application.
 app.include_router(agents)
 ```
 
@@ -141,7 +142,6 @@ The pre-built images support the following environment variables for customisati
 
 | Variable           | Example Value      | Description                                                |
 |--------------------|--------------------|------------------------------------------------------------|
-| `AGENT_FRAMEWORK`  | `pydantic-ai`     | Specifies the agent framework to use.                      |
 | `AGENT_MODULE`     | `agent.pydantic_ai` | Path to the agent module.                                  |
 | `AGENT_CLASS`      | `agent`           | Class name for the agent.                                  |
 | `SECURITY_MODULE`  | `agent.pydantic_ai` | Specifies the security module for the agent. |

docker/main.py

@@ -4,7 +4,6 @@
 import importlib
 
 # get environment variables
-agent_framework = os.getenv("AGENT_FRAMEWORK", "pydantic_ai")
 agent_module = os.getenv("AGENT_MODULE", "agent")
 agent_class = os.getenv("AGENT_CLASS", "agent")
 security_module = os.getenv("SECURITY_MODULE", None)
@@ -19,22 +18,6 @@
 if security_class == '':
     security_class = None
 
-# dynamically import agent framework
-if agent_framework == "pydantic-ai":
-    from fastapi_agents.pydantic_ai import PydanticAIAgent as AgentWrapper
-    from pydantic_ai import Agent as BaseAgentClass
-elif agent_framework == "smolagents":
-    from fastapi_agents.smolagents import SmolagentsAgent as AgentWrapper
-    from smolagents import MultiStepAgent as BaseAgentClass
-elif agent_framework == "llama-index":
-    from fastapi_agents.llama_index import LlamaIndexAgent as AgentWrapper
-    from llama_index.core.agent.types import BaseAgent as BaseAgentClass
-elif agent_framework == "crewai":
-    from fastapi_agents.crewai import CrewAIAgent as AgentWrapper
-    from crewai import Crew as BaseAgentClass
-else:
-    raise ValueError(f"Unknown agent framework: {agent_framework}")
-
 # create FastAPI app and FastAPIAgents instance
 app = FastAPI()
 agents = FastAPIAgents(path_prefix=api_prefix, mode=api_mode)
@@ -43,10 +26,6 @@
 import_agent_module = importlib.import_module(agent_module)
 agent = getattr(import_agent_module, agent_class)
 
-# check type of agent
-if not issubclass(type(agent), BaseAgentClass):
-    raise ValueError(f"Agent must be a subclass of {type(BaseAgentClass)}, not {agent}")
-
 # dynamically import security module
 if security_class and security_module:
     import_security_module = importlib.import_module(security_module)
@@ -59,7 +38,7 @@
 # register agent with FastAPIAgents
 agents.register(
     api_endpoint, 
-    AgentWrapper(agent), 
+    agent, 
     security_dependency=security
 )
 app.include_router(agents)