pydantic-rpc 0.2.0__tar.gz → 0.2.1__tar.gz

{pydantic_rpc-0.2.0 → pydantic_rpc-0.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: pydantic-rpc
-Version: 0.2.0
+Version: 0.2.1
 Summary: A Python library for building gRPC/ConnectRPC services with Pydantic models.
 Author: Yasushi Itoh
 Requires-Python: >=3.11
@@ -14,10 +14,10 @@ Description-Content-Type: text/markdown
 
 # 🚀 PydanticRPC
 
-**PydanticRPC** is a Python library that enables you to rapidly expose Pydantic models via gRPC/ConnectRPC services without writing any protobuf files. Instead, it automatically generates protobuf files on the fly from the method signatures of your Python objects and the type signatures of your Pydantic models.
+**PydanticRPC** is a Python library that enables you to rapidly expose [Pydantic](https://docs.pydantic.dev/) models via [gRPC](https://grpc.io/)/[ConnectRPC](https://connectrpc.com/docs/protocol/) services without writing any protobuf files. Instead, it automatically generates protobuf files on the fly from the method signatures of your Python objects and the type signatures of your Pydantic models.
 
 
-Below is an example of a simple gRPC service that exposes a PydanticAI agent:
+Below is an example of a simple gRPC service that exposes a [PydanticAI](https://ai.pydantic.dev/) agent:
 
 ```python
 import asyncio
@@ -54,6 +54,40 @@ if __name__ == "__main__":
     loop.run_until_complete(s.run(OlympicsLocationAgent()))
 ```
 
+And here is an example of a simple ConnectRPC service that exposes the same agent as an ASGI application:
+
+```python
+import asyncio
+
+from pydantic_ai import Agent
+from pydantic_rpc import ConnecpyASGIApp, Message
+
+
+class CityLocation(Message):
+    city: str
+    country: str
+
+
+class Olympics(Message):
+    year: int
+
+    def prompt(self):
+        return f"Where were the Olympics held in {self.year}?"
+
+
+class OlympicsLocationAgent:
+    def __init__(self):
+        self._agent = Agent("ollama:llama3.2", result_type=CityLocation)
+
+    async def ask(self, req: Olympics) -> CityLocation:
+        result = await self._agent.run(req.prompt())
+        return result.data
+
+app = ConnecpyASGIApp()
+app.mount(OlympicsLocationAgent())
+
+```
+
 
 ## 💡 Key Features
 

{pydantic_rpc-0.2.0 → pydantic_rpc-0.2.1}/README.md

@@ -1,9 +1,9 @@
 # 🚀 PydanticRPC
 
-**PydanticRPC** is a Python library that enables you to rapidly expose Pydantic models via gRPC/ConnectRPC services without writing any protobuf files. Instead, it automatically generates protobuf files on the fly from the method signatures of your Python objects and the type signatures of your Pydantic models.
+**PydanticRPC** is a Python library that enables you to rapidly expose [Pydantic](https://docs.pydantic.dev/) models via [gRPC](https://grpc.io/)/[ConnectRPC](https://connectrpc.com/docs/protocol/) services without writing any protobuf files. Instead, it automatically generates protobuf files on the fly from the method signatures of your Python objects and the type signatures of your Pydantic models.
 
 
-Below is an example of a simple gRPC service that exposes a PydanticAI agent:
+Below is an example of a simple gRPC service that exposes a [PydanticAI](https://ai.pydantic.dev/) agent:
 
 ```python
 import asyncio
@@ -40,6 +40,40 @@ if __name__ == "__main__":
     loop.run_until_complete(s.run(OlympicsLocationAgent()))
 ```
 
+And here is an example of a simple ConnectRPC service that exposes the same agent as an ASGI application:
+
+```python
+import asyncio
+
+from pydantic_ai import Agent
+from pydantic_rpc import ConnecpyASGIApp, Message
+
+
+class CityLocation(Message):
+    city: str
+    country: str
+
+
+class Olympics(Message):
+    year: int
+
+    def prompt(self):
+        return f"Where were the Olympics held in {self.year}?"
+
+
+class OlympicsLocationAgent:
+    def __init__(self):
+        self._agent = Agent("ollama:llama3.2", result_type=CityLocation)
+
+    async def ask(self, req: Olympics) -> CityLocation:
+        result = await self._agent.run(req.prompt())
+        return result.data
+
+app = ConnecpyASGIApp()
+app.mount(OlympicsLocationAgent())
+
+```
+
 
 ## 💡 Key Features
 

{pydantic_rpc-0.2.0 → pydantic_rpc-0.2.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "pydantic-rpc"
-version = "0.2.0"
+version = "0.2.1"
 description = "A Python library for building gRPC/ConnectRPC services with Pydantic models."
 authors = [
     { name = "Yasushi Itoh" }

{pydantic_rpc-0.2.0 → pydantic_rpc-0.2.1}/src/pydantic_rpc/core.py

@@ -865,7 +865,11 @@ def generate_and_compile_proto(obj: object, package_name: str = ""):
         pb2_grpc_module = importlib.import_module(
             f"{obj.__class__.__name__.lower()}_pb2_grpc"
         )
-        return pb2_grpc_module, pb2_module
+
+        if pb2_grpc_module is not None and pb2_module is not None:
+            return pb2_grpc_module, pb2_module
+
+        # If the modules are not found, generate and compile the proto files.
 
     klass = obj.__class__
     proto_file = generate_proto(obj, package_name)
@@ -892,7 +896,11 @@ def generate_and_compile_proto_using_connecpy(obj: object, package_name: str = "
         connecpy_module = importlib.import_module(
             f"{obj.__class__.__name__.lower()}_connecpy"
         )
-        return connecpy_module, pb2_module
+
+        if connecpy_module is not None and pb2_module is not None:
+            return connecpy_module, pb2_module
+
+        # If the modules are not found, generate and compile the proto files.
 
     klass = obj.__class__
     proto_file = generate_proto(obj, package_name)