AI Agent Factory with Claude Code Subagents

use-cases/agent-factory-with-subagents/agents/rag_agent/agent.py

@@ -0,0 +1,21 @@
+"""Main agent implementation for Semantic Search."""
+
+from pydantic_ai import Agent, RunContext
+from typing import Any
+
+from providers import get_llm_model
+from dependencies import AgentDependencies
+from prompts import MAIN_SYSTEM_PROMPT
+from tools import semantic_search, hybrid_search
+
+
+# Initialize the semantic search agent
+search_agent = Agent(
+    get_llm_model(),
+    deps_type=AgentDependencies,
+    system_prompt=MAIN_SYSTEM_PROMPT
+)
+
+# Register search tools
+search_agent.tool(semantic_search)
+search_agent.tool(hybrid_search)