update for pydanticai breaking change

nicucalcea · nicucalcea · commit 69391cba319d · 2025-11-06T17:40:46.000Z
diff --git a/augmenta/agent.py b/augmenta/agent.py
@@ -163,7 +163,7 @@ async def run(
             )
 
             # Return the appropriate result format
-            return result.data.model_dump() if response_format else result.data
+            return result.output.model_dump() if response_format else result.output
         except Exception as e:
             logfire.error(f"LLM request failed: {e}")
             raise RuntimeError(f"LLM request failed: {e}")
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "augmenta"
-version = "0.1.1"
+version = "0.1.2"
 description = "AI-powered library for data enrichment using search results"
 readme = "README.md"
 requires-python = ">=3.10"
@@ -15,7 +15,7 @@ dependencies = [
     "pyyaml>=6.0.0",
     "pandas>=1.0.0",
     "pydantic>=2.0.0",
-    "pydantic-ai>=0.4.6",
+    "pydantic-ai>=0.6.0",
     "logfire[httpx]>=3.9.0",
     "trafilatura>=2.0.0",
     "aiolimiter>=1.2.0",

Original file line number	Diff line number	Diff line change
`@@ -163,7 +163,7 @@ async def run(`
`163`	`163`	`)`
`164`	`164`
`165`	`165`	`# Return the appropriate result format`
`166`		`- return result.data.model_dump() if response_format else result.data`
	`166`	`+ return result.output.model_dump() if response_format else result.output`
`167`	`167`	`except Exception as e:`
`168`	`168`	`logfire.error(f"LLM request failed: {e}")`
`169`	`169`	`raise RuntimeError(f"LLM request failed: {e}")`