feat: added agentkit implimentation

Author: iamprashant

README.md

@@ -1,6 +1,6 @@
 # Rapida Python SDK
 
-[![Build and Publish Python SDK](https://github.com/rapidaai/rapida-python/actions/workflows/python-build.yml/badge.svg?branch=main)](https://github.com/rapidaai/rapida-python/actions/workflows/build.yml)
+[![Build and Publish Python SDK](https://github.com/rapidaai/rapida-python/actions/workflows/python-build.yml/badge.svg?branch=main)](https://github.com/rapidaai/rapida-python/actions/workflows/python-build.yml)
 
 The Rapida Python SDK provides a powerful interface for interacting with Rapida AI services. This SDK simplifies the process of making API calls, handling authentication, and managing responses from Rapida endpoints.
 
@@ -17,18 +17,13 @@ pip install rapida-python
 Here's how to get started with the Rapida Python SDK:
 
 ```python
-from rapida import (
-    ConnectionConfig,
-)
-from pprint import pprint
-
+from rapida import ConnectionConfig
 
 connection_config = ConnectionConfig.default_connection_config(
     ConnectionConfig.with_sdk(
-        "{your-api-key-here}"  # Replace with your actual API key
+        "{your-api-key-here}"
     )
 )
-
 ```
 
 ## Authentication
@@ -40,8 +35,11 @@ You can configure the Rapida SDK to authenticate using your **API Key** or **Per
 ```python
 import os
 from rapida.connections import ConnectionConfig
-connection_config = ConnectionConfig.with_sdk(
-    os.environ["RAPIDA_API_KEY"]  # API Key from environment variables
+
+connection_config = ConnectionConfig.default_connection_config(
+    ConnectionConfig.with_sdk(
+        os.environ["RAPIDA_API_KEY"]  # API Key from environment variables
+    )
 )
 ```
 
@@ -51,25 +49,46 @@ connection_config = ConnectionConfig.with_sdk(
 import os
 from rapida.connections import ConnectionConfig
 
-connection_config = ConnectionConfig.with_personal_token(
-  os.environ["RAPIDA_AUTHORIZATION_TOKEN"],  # Authorization Token
-  os.environ["RAPIDA_AUTH_ID"],              # Authentication ID
-  os.environ["RAPIDA_PROJECT_ID"],           # Project ID
+connection_config = ConnectionConfig.default_connection_config(
+    ConnectionConfig.with_personal_token(
+        os.environ["RAPIDA_AUTHORIZATION_TOKEN"],  # Authorization Token
+        os.environ["RAPIDA_AUTH_ID"],              # Authentication ID
+        os.environ["RAPIDA_PROJECT_ID"],           # Project ID
+    )
 )
 ```
 
 ## Configuration Options
 
-The `DefaultConnectionConfig` accepts multiple options for configuring the SDK. Key options include:
+`ConnectionConfig` accepts multiple options for configuring the SDK:
 
 - `with_sdk(api_key: str)`: Sets the API key for authentication.
 - `with_personal_token(auth_token: str, auth_id: str, project_id: str)`: Configures the connection for personal tokens.
-- `with_endpoint_url(url: str)`: Overrides the default API endpoint URL.
-- `with_timeout(timeout: float)`: Sets the timeout for API requests (in seconds).
+- `with_webplugin_client(api_key: str, user_id: Optional[str] = None)`: Configures web plugin client authentication.
+- `with_debugger(authorization: str, user_id: str, project_id: str)`: Configures debugger authentication.
+- `with_custom_endpoint(endpoint: Optional[dict] = None, debug: Optional[bool] = None)`: Overrides the default assistant, web, and endpoint API hosts.
+- `with_local()`: Uses local service endpoints and insecure gRPC channels for local development.
+- `default_connection_config(auth)`: Creates a `ConnectionConfig` with the supplied auth metadata.
+
+Example using custom endpoints:
+
+```python
+from rapida import ConnectionConfig
+
+connection_config = ConnectionConfig.default_connection_config(
+    ConnectionConfig.with_sdk("{your-api-key-here}")
+).with_custom_endpoint(
+    {
+        "assistant": "assistant.example.com:50051",
+        "web": "api.example.com:50051",
+        "endpoint": "endpoint.example.com:50051",
+    }
+)
+```
 
 ## Compatibility
 
-This SDK requires **Python 3.8 or later**. Ensure your system meets this requirement:
+This SDK requires **Python 3.9 or later**. Ensure your system meets this requirement:
 
 ```bash
 python --version

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "rapida-python"
-version = "0.1.28"
+version = "0.1.29"
 description = "RapidaAI SDK to integrate rapida.ai APIs"
 readme = "README.md"
 authors = [{name = "RapidaAI", email = "code@rapida.ai"}]

rapida/agentkit/__init__.py

@@ -171,12 +171,13 @@ def intercept_service(self, continuation, handler_call_details):
         return continuation(handler_call_details)
 
     def _abort_handler(self):
-        def abort(ignored_request, context):
+        def abort(ignored_request_iterator, context):
             context.abort(
                 grpc.StatusCode.UNAUTHENTICATED, "Invalid authorization token"
             )
+            return iter(())
 
-        return grpc.unary_unary_rpc_method_handler(abort)
+        return grpc.stream_stream_rpc_method_handler(abort)
 
 
 # ============================================================================
@@ -658,6 +659,7 @@ def stop(self, grace: Optional[int] = 5) -> None:
         if self._server:
             logger.info(f"Stopping server (grace={grace}s)...")
             self._server.stop(grace)
+            self._server = None
             logger.info("Server stopped")
 
     def wait_for_termination(self, timeout: Optional[float] = None) -> bool:

tests/agentkit/test_agentkit.py

@@ -895,6 +895,12 @@ def test_abort_handler_is_not_none(self):
         handler = interceptor._abort_handler()
         assert handler is not None
 
+    def test_abort_handler_is_stream_stream_compatible(self):
+        interceptor = AuthorizationInterceptor(AuthConfig(enabled=True, token="t"))
+        handler = interceptor._abort_handler()
+        assert handler.request_streaming is True
+        assert handler.response_streaming is True
+
 
 # ============================================================================
 # AgentKitServer
@@ -1123,6 +1129,12 @@ def test_stop_with_default_grace(self):
         server.stop()
         mock_grpc_server.stop.assert_called_once_with(5)
 
+    def test_stop_marks_server_not_running(self):
+        server = self._make_server()
+        server._server = MagicMock()
+        server.stop()
+        assert server.is_running is False
+
     def test_stop_noop_when_server_never_started(self):
         server = self._make_server()
         # Must not raise