Add llama_index framework (#46)

* Add llama_index framework

* Fix missing `main_agent`

examples/llama_index_single_agent.yaml

@@ -0,0 +1,8 @@
+location: Pontevedra
+date: 2025-04-02 12:00
+max_driving_hours: 2
+
+framework: llama_index
+
+main_agent:
+  model_id: o3-mini

src/surf_spot_finder/cli.py

@@ -29,9 +29,9 @@ def find_surf_spot(
     config = Config.model_validate(yaml.safe_load(Path(config_file).read_text()))
 
     if not config.main_agent.instructions:
-        if config.main_agent.agent_framework == AgentFramework.SMOLAGENTS:
+        if config.framework == AgentFramework.SMOLAGENTS:
             config.main_agent.instructions = SYSTEM_PROMPT
-        elif config.main_agent.agent_framework == AgentFramework.OPENAI:
+        elif config.framework == AgentFramework.OPENAI:
             config.main_agent.instructions = SINGLE_AGENT_SYSTEM_PROMPT
 
     logger.info("Setting up tracing")

src/surf_spot_finder/config.py

@@ -1,7 +1,7 @@
 from typing import Annotated
 
 from any_agent import AgentFramework
-from any_agent.schema import AgentSchema
+from any_agent.config import AgentConfig
 from pydantic import AfterValidator, BaseModel, ConfigDict, FutureDatetime, PositiveInt
 import yaml
 
@@ -32,8 +32,8 @@ class Config(BaseModel):
 
     framework: AgentFramework
 
-    main_agent: AgentSchema
-    managed_agents: list[AgentSchema] | None = None
+    main_agent: AgentConfig
+    managed_agents: list[AgentConfig] | None = None
 
     @classmethod
     def from_yaml(cls, yaml_path: str) -> "Config":