fix readme

Author: ponytailer

README.md

@@ -73,11 +73,9 @@ class MyAPIClient(RequestsWebClient):
         pass
 
     @delete("/users", agno_tool=True, tool_description="this is the function to delete user")
-    def delete_user(self, user_id: str):
+    def delete_user(self, user_id: str, request_headers: Dict[str, Any]):
         ...
 
-session = requests.Session()
-
 
 # Use the client
 client = MyAPIClient(base_url="https://localhost")
@@ -86,6 +84,10 @@ user = client.get_user(user_id=123)
 user_body = CreateUser(name="john", email="123@gmail.com")
 user = client.create_user(user_body)
 
+# will update the client headers.
+client.delete_user("123", {"ba": "your"})
+
+
 from agno.agent import Agent
 
 agent = Agent(.....)
@@ -170,6 +172,7 @@ You can initialize clients with custom configurations:
 client = MyAPIClient(
     base_url="https://api.example.com",
     headers={"Custom-Header": "value"},
+    session=requests.Session(),  # your own session
     timeout=30  # in seconds
 )
 

pydantic_client/async_client.py

@@ -1,4 +1,4 @@
-from typing import Any, Dict, Optional, Type, TypeVar
+from typing import Any, Dict, Optional, TypeVar
 
 from pydantic import BaseModel
 
@@ -39,7 +39,7 @@ async def _request(self, request_info: RequestInfo) -> Any:
                 data = await response.json()
 
                 if response_model is not None:
-                    return response_model.model_validate(data)
+                    return response_model.model_validate(data, from_attributes=True)
                 return data
 
 
@@ -71,5 +71,5 @@ async def _request(self, request_info: RequestInfo) -> Any:
             response.raise_for_status()
             data = response.json()
             if response_model is not None:
-                return response_model.model_validate(data)
+                return response_model.model_validate(data, from_attributes=True)
             return data

pydantic_client/base.py

@@ -4,7 +4,7 @@
 import statsd
 
 from abc import ABC, abstractmethod
-from typing import Any, Dict, Optional, Type, TypeVar, List
+from typing import Any, Dict, Optional, TypeVar, List
 
 from pydantic import BaseModel
 from .schema import RequestInfo

pydantic_client/decorators.py

@@ -1,6 +1,6 @@
 import inspect
 from functools import wraps
-from typing import Any, Callable, Dict, Optional
+from typing import Callable, Optional
 
 from pydantic import BaseModel
 
@@ -35,6 +35,7 @@ def _process_request_params(
 
     # Get return type for response model
     return_type = sig.return_annotation
+
     response_model = (
         return_type
         if isinstance(return_type, type) and issubclass(return_type, BaseModel)

pydantic_client/schema.py

@@ -1,4 +1,4 @@
-from typing import Dict, Optional, Any, Type
+from typing import Dict, Optional, Any
 
 from pydantic import BaseModel
 
@@ -10,4 +10,4 @@ class RequestInfo(BaseModel):
     json: Optional[Dict[str, Any]] = None
     data: Optional[Dict[str, Any]] = None
     headers: Optional[Dict[str, Any]] = {}
-    response_model: Optional[Type[BaseModel]] = None
+    response_model: Optional[Any] = None

pydantic_client/sync_client.py

@@ -1,4 +1,4 @@
-from typing import Any, Dict, Optional, Type, TypeVar
+from typing import Any, Dict, Optional, TypeVar
 
 import requests
 from pydantic import BaseModel
@@ -29,7 +29,8 @@ def _request(self, request_info: RequestInfo) -> Any:
 
         response = requests.request(**request_params, timeout=self.timeout)
         response.raise_for_status()
+
         data = response.json()
         if response_model is not None:
-            return response_model.model_validate(data)
+            return response_model.model_validate(data, from_attributes=True)
         return data