Merge pull request #121 from JoshuaC215/ddg-update

Bump DDG-search, fix typing
JoshuaC215 · Dec 12, 2024 · d465541 · d465541
2 parents c72f488 + 71e310b
commit d465541
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 204 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -16,7 +16,7 @@ classifiers = [
 requires-python = ">=3.11, <3.13"
 
 dependencies = [
-    "duckduckgo-search>=6.3.6",
+    "duckduckgo-search>=6.4.1",
     "fastapi ~=0.115.5",
     "httpx ~=0.27.2",
     "langchain-core ~=0.3.20",

diff --git a/src/schema/schema.py b/src/schema/schema.py
@@ -3,7 +3,7 @@
 from pydantic import BaseModel, Field, SerializeAsAny
 from typing_extensions import TypedDict
 
-from schema.models import AllModelEnum
+from schema.models import AllModelEnum, AnthropicModelName, OpenAIModelName
 
 
 class AgentInfo(BaseModel):
@@ -47,8 +47,8 @@ class UserInput(BaseModel):
     model: SerializeAsAny[AllModelEnum] | None = Field(
         title="Model",
         description="LLM Model to use for the agent.",
-        default="gpt-4o-mini",
-        examples=["gpt-4o-mini", "llama-3.1-70b"],
+        default=OpenAIModelName.GPT_4O_MINI,
+        examples=[OpenAIModelName.GPT_4O_MINI, AnthropicModelName.HAIKU_35],
     )
     thread_id: str | None = Field(
         description="Thread ID to persist and continue a multi-turn conversation.",