mirror of
https://github.com/coleam00/context-engineering-intro.git
synced 2025-12-18 02:05:25 +00:00
22 lines
536 B
Python
22 lines
536 B
Python
|
"""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)
|