code-puppy 0.0.192__tar.gz → 0.0.193__tar.gz

{code_puppy-0.0.192 → code_puppy-0.0.193}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: code-puppy
-Version: 0.0.192
+Version: 0.0.193
 Summary: Code generation agent
 Project-URL: repository, https://github.com/mpfaffenberger/code_puppy
 Project-URL: HomePage, https://github.com/mpfaffenberger/code_puppy
@@ -26,7 +26,7 @@ Requires-Dist: openai>=1.99.1
 Requires-Dist: pathspec>=0.11.0
 Requires-Dist: playwright>=1.40.0
 Requires-Dist: prompt-toolkit>=3.0.38
-Requires-Dist: pydantic-ai<=0.8
+Requires-Dist: pydantic-ai==1.0.6
 Requires-Dist: pydantic>=2.4.0
 Requires-Dist: pyjwt>=2.8.0
 Requires-Dist: pytest-cov>=6.1.1

{code_puppy-0.0.192 → code_puppy-0.0.193}/code_puppy/agents/base_agent.py

@@ -22,7 +22,7 @@ from pydantic_ai.messages import (
     ToolReturn,
     ToolReturnPart,
 )
-from pydantic_ai.models.openai import OpenAIModelSettings
+from pydantic_ai.models.openai import OpenAIChatModelSettings
 from pydantic_ai.settings import ModelSettings
 
 # Consolidated relative imports
@@ -835,7 +835,7 @@ class BaseAgent(ABC):
                 get_openai_reasoning_effort()
             )
             model_settings_dict["extra_body"] = {"verbosity": "low"}
-            model_settings = OpenAIModelSettings(**model_settings_dict)
+            model_settings = OpenAIChatModelSettings(**model_settings_dict)
 
         self.cur_model = model
         p_agent = PydanticAgent(

{code_puppy-0.0.192 → code_puppy-0.0.193}/code_puppy/model_factory.py

@@ -8,11 +8,11 @@ import httpx
 from anthropic import AsyncAnthropic
 from openai import AsyncAzureOpenAI
 from pydantic_ai.models.anthropic import AnthropicModel
-from pydantic_ai.models.gemini import GeminiModel
+from pydantic_ai.models.google import GoogleModel
 from pydantic_ai.models.openai import OpenAIChatModel
 from pydantic_ai.providers.anthropic import AnthropicProvider
 from pydantic_ai.providers.cerebras import CerebrasProvider
-from pydantic_ai.providers.google_gla import GoogleGLAProvider
+from pydantic_ai.providers.google import GoogleProvider
 from pydantic_ai.providers.openai import OpenAIProvider
 from pydantic_ai.providers.openrouter import OpenRouterProvider
 
@@ -142,9 +142,9 @@ class ModelFactory:
         model_type = model_config.get("type")
 
         if model_type == "gemini":
-            provider = GoogleGLAProvider(api_key=os.environ.get("GEMINI_API_KEY", ""))
+            provider = GoogleProvider(api_key=os.environ.get("GEMINI_API_KEY", ""))
 
-            model = GeminiModel(model_name=model_config["name"], provider=provider)
+            model = GoogleModel(model_name=model_config["name"], provider=provider)
             setattr(model, "provider", provider)
             return model
 
@@ -266,7 +266,7 @@ class ModelFactory:
             url, headers, verify, api_key = get_custom_config(model_config)
             os.environ["GEMINI_API_KEY"] = api_key
 
-            class CustomGoogleGLAProvider(GoogleGLAProvider):
+            class CustomGoogleGLAProvider(GoogleProvider):
                 def __init__(self, *args, **kwargs):
                     super().__init__(*args, **kwargs)
 
@@ -281,7 +281,7 @@ class ModelFactory:
                     return _client
 
             google_gla = CustomGoogleGLAProvider(api_key=api_key)
-            model = GeminiModel(model_name=model_config["name"], provider=google_gla)
+            model = GoogleModel(model_name=model_config["name"], provider=google_gla)
             return model
         elif model_type == "cerebras":
             url, headers, verify, api_key = get_custom_config(model_config)

{code_puppy-0.0.192 → code_puppy-0.0.193}/pyproject.toml

@@ -4,12 +4,12 @@ build-backend = "hatchling.build"
 
 [project]
 name = "code-puppy"
-version = "0.0.192"
+version = "0.0.193"
 description = "Code generation agent"
 readme = "README.md"
 requires-python = ">=3.11"
 dependencies = [
-    "pydantic-ai<=0.8",
+    "pydantic-ai==1.0.6",
     "httpx>=0.24.1",
     "rich>=13.4.2",
     "logfire>=0.7.1",