feat: added agentkit test and update protos

Author: iamprashant

README.md

@@ -86,4 +86,3 @@ pip install --upgrade rapida-python
 ## Conclusion
 
 The Rapida Python SDK provides everything necessary to integrate seamlessly with Rapida AI services, offering flexible configuration and authentication options. With the examples provided, you should be able to get started quickly and make advanced API calls as needed.
-

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "rapida-python"
-version = "0.1.25"
+version = "0.1.26"
 description = "RapidaAI SDK to integrate rapida.ai APIs"
 readme = "README.md"
 authors = [{name = "RapidaAI", email = "code@rapida.ai"}]
@@ -58,4 +58,4 @@ packages = [
 "rapida.clients.protos" = ["*.py"]
 
 [tool.setuptools.exclude-package-data]
-"*" = ["tests", "examples"]
+"*" = ["tests", "examples"]

rapida/agentkit/__init__.py

@@ -72,7 +72,7 @@ def Talk(self, request_iterator, context):
 import os
 from concurrent import futures
 from dataclasses import dataclass
-from typing import Any, Callable, Dict, Iterator, Optional
+from typing import Any, Callable, Dict, Optional
 
 import grpc
 from grpc import ServerInterceptor
@@ -176,6 +176,7 @@ def abort(ignored_request, context):
             context.abort(
                 grpc.StatusCode.UNAUTHENTICATED, "Invalid authorization token"
             )
+
         return grpc.unary_unary_rpc_method_handler(abort)
 
 
@@ -223,9 +224,7 @@ def Talk(self, request_iterator, context):
     # RESPONSE BUILDERS - Send data back to Rapida
     # ========================================================================
 
-    def response(
-        self, code: int = 200, success: bool = True, **kwargs
-    ) -> TalkOutput:
+    def response(self, code: int = 200, success: bool = True, **kwargs) -> TalkOutput:
         """
         Build a generic response to send back to Rapida.
 
@@ -338,13 +337,12 @@ def tool_call(
         for k, v in (args or {}).items():
             _args[str(k)] = string_to_any(str(v))
 
-        return self.response(tool=ConversationToolCall(
-            id=str(msg_id),
-            toolId=str(tool_id),
-            name=str(name),
-            args=_args
-        ))
-    
+        return self.response(
+            tool=ConversationToolCall(
+                id=str(msg_id), toolId=str(tool_id), name=str(name), args=_args
+            )
+        )
+
     def tool_call_result(
         self, msg_id: str, tool_id: str, name: str, result: Any, success: bool = True
     ) -> TalkOutput:
@@ -373,17 +371,17 @@ def tool_call_result(
                 # For non-dict results, store under "result" key
                 _args["result"] = string_to_any(str(result))
 
-        return self.response(toolResult=ConversationToolResult(
-            id=str(msg_id),
-            toolId=str(tool_id),
-            name=str(name),
-            success=bool(success),
-            args=_args
-        ))
+        return self.response(
+            toolResult=ConversationToolResult(
+                id=str(msg_id),
+                toolId=str(tool_id),
+                name=str(name),
+                success=bool(success),
+                args=_args,
+            )
+        )
 
-    def transfer_call(
-        self, msg_id: str, args: Dict[str, Any]
-    ) -> TalkOutput:
+    def transfer_call(self, msg_id: str, args: Dict[str, Any]) -> TalkOutput:
         """
         Send a transfer call directive back to Rapida.
 
@@ -399,15 +397,15 @@ def transfer_call(
         for k, v in (args or {}).items():
             _args[str(k)] = string_to_any(str(v))
 
-        return self.response(directive=ConversationDirective(
-            id=str(msg_id),
-            type=ConversationDirective.TRANSFER_CONVERSATION,
-            args=_args
-        ))
+        return self.response(
+            directive=ConversationDirective(
+                id=str(msg_id),
+                type=ConversationDirective.TRANSFER_CONVERSATION,
+                args=_args,
+            )
+        )
 
-    def terminate_call(
-        self, msg_id: str, args: Dict[str, Any]
-    ) -> TalkOutput:
+    def terminate_call(self, msg_id: str, args: Dict[str, Any]) -> TalkOutput:
         """
         Send a tool call that ends the conversation back to Rapida.
 
@@ -424,11 +422,12 @@ def terminate_call(
         # Set map fields with Any values after construction
         for k, v in (args or {}).items():
             _arg[str(k)] = string_to_any(str(v))
-        
-        return self.response(directive=ConversationDirective(
-            id=str(msg_id),
-            type=ConversationDirective.END_CONVERSATION,
-            args=_arg))
+
+        return self.response(
+            directive=ConversationDirective(
+                id=str(msg_id), type=ConversationDirective.END_CONVERSATION, args=_arg
+            )
+        )
 
     # ========================================================================
     # REQUEST HELPERS - Receive data from Rapida
@@ -486,7 +485,9 @@ def get_assistant_id(self, request: TalkInput) -> Optional[int]:
         Returns:
             Assistant ID, or None if not an initialization request or assistant is unset
         """
-        if request.HasField("initialization") and request.initialization.HasField("assistant"):
+        if request.HasField("initialization") and request.initialization.HasField(
+            "assistant"
+        ):
             return request.initialization.assistant.assistantId
         return None
 
@@ -504,17 +505,11 @@ def is_message_request(self, request: TalkInput) -> bool:
 
     def is_text_message(self, request: TalkInput) -> bool:
         """Check if request is a text message."""
-        return (
-            request.HasField("message")
-            and request.message.HasField("text")
-        )
+        return request.HasField("message") and request.message.HasField("text")
 
     def is_audio_message(self, request: TalkInput) -> bool:
         """Check if request is an audio message."""
-        return (
-            request.HasField("message")
-            and request.message.HasField("audio")
-        )
+        return request.HasField("message") and request.message.HasField("audio")
 
 
 # ============================================================================

rapida/clients/__init__.py

@@ -18,4 +18,4 @@
 #  OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 #  THE SOFTWARE.
 #
-#  Author: Prashant <prashant@rapida.ai>
+#  Author: Prashant <prashant@rapida.ai>

rapida/clients/protos/agentkit_pb2_grpc.py

@@ -5,7 +5,7 @@
 
 import rapida.clients.protos.agentkit_pb2 as agentkit__pb2
 
-GRPC_GENERATED_VERSION = '1.78.1'
+GRPC_GENERATED_VERSION = '1.78.0'
 GRPC_VERSION = grpc.__version__
 _version_not_supported = False
 

rapida/clients/protos/artifacts

@@ -4,7 +4,7 @@
 import warnings
 
 
-GRPC_GENERATED_VERSION = '1.78.1'
+GRPC_GENERATED_VERSION = '1.78.0'
 GRPC_VERSION = grpc.__version__
 _version_not_supported = False
 

rapida/clients/protos/assistant_analysis_pb2_grpc.py

@@ -152,6 +152,61 @@ class GetAllAssistantTelemetryRequest(_message.Message):
     assistant: _common_pb2.AssistantDefinition
     def __init__(self, paginate: _Optional[_Union[_common_pb2.Paginate, _Mapping]] = ..., criterias: _Optional[_Iterable[_Union[_common_pb2.Criteria, _Mapping]]] = ..., assistant: _Optional[_Union[_common_pb2.AssistantDefinition, _Mapping]] = ...) -> None: ...
 
+class TelemetryEvent(_message.Message):
+    __slots__ = ("messageId", "assistantId", "assistantConversationId", "projectId", "organizationId", "name", "data", "time")
+    class DataEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: str
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
+    MESSAGEID_FIELD_NUMBER: _ClassVar[int]
+    ASSISTANTID_FIELD_NUMBER: _ClassVar[int]
+    ASSISTANTCONVERSATIONID_FIELD_NUMBER: _ClassVar[int]
+    PROJECTID_FIELD_NUMBER: _ClassVar[int]
+    ORGANIZATIONID_FIELD_NUMBER: _ClassVar[int]
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    DATA_FIELD_NUMBER: _ClassVar[int]
+    TIME_FIELD_NUMBER: _ClassVar[int]
+    messageId: str
+    assistantId: int
+    assistantConversationId: int
+    projectId: int
+    organizationId: int
+    name: str
+    data: _containers.ScalarMap[str, str]
+    time: _timestamp_pb2.Timestamp
+    def __init__(self, messageId: _Optional[str] = ..., assistantId: _Optional[int] = ..., assistantConversationId: _Optional[int] = ..., projectId: _Optional[int] = ..., organizationId: _Optional[int] = ..., name: _Optional[str] = ..., data: _Optional[_Mapping[str, str]] = ..., time: _Optional[_Union[datetime.datetime, _timestamp_pb2.Timestamp, _Mapping]] = ...) -> None: ...
+
+class TelemetryMetric(_message.Message):
+    __slots__ = ("contextId", "assistantId", "assistantConversationId", "projectId", "organizationId", "scope", "metrics", "time")
+    CONTEXTID_FIELD_NUMBER: _ClassVar[int]
+    ASSISTANTID_FIELD_NUMBER: _ClassVar[int]
+    ASSISTANTCONVERSATIONID_FIELD_NUMBER: _ClassVar[int]
+    PROJECTID_FIELD_NUMBER: _ClassVar[int]
+    ORGANIZATIONID_FIELD_NUMBER: _ClassVar[int]
+    SCOPE_FIELD_NUMBER: _ClassVar[int]
+    METRICS_FIELD_NUMBER: _ClassVar[int]
+    TIME_FIELD_NUMBER: _ClassVar[int]
+    contextId: str
+    assistantId: int
+    assistantConversationId: int
+    projectId: int
+    organizationId: int
+    scope: str
+    metrics: _containers.RepeatedCompositeFieldContainer[_common_pb2.Metric]
+    time: _timestamp_pb2.Timestamp
+    def __init__(self, contextId: _Optional[str] = ..., assistantId: _Optional[int] = ..., assistantConversationId: _Optional[int] = ..., projectId: _Optional[int] = ..., organizationId: _Optional[int] = ..., scope: _Optional[str] = ..., metrics: _Optional[_Iterable[_Union[_common_pb2.Metric, _Mapping]]] = ..., time: _Optional[_Union[datetime.datetime, _timestamp_pb2.Timestamp, _Mapping]] = ...) -> None: ...
+
+class TelemetryRecord(_message.Message):
+    __slots__ = ("event", "metric")
+    EVENT_FIELD_NUMBER: _ClassVar[int]
+    METRIC_FIELD_NUMBER: _ClassVar[int]
+    event: TelemetryEvent
+    metric: TelemetryMetric
+    def __init__(self, event: _Optional[_Union[TelemetryEvent, _Mapping]] = ..., metric: _Optional[_Union[TelemetryMetric, _Mapping]] = ...) -> None: ...
+
 class GetAllAssistantTelemetryResponse(_message.Message):
     __slots__ = ("code", "success", "data", "error", "paginated")
     CODE_FIELD_NUMBER: _ClassVar[int]
@@ -161,10 +216,10 @@ class GetAllAssistantTelemetryResponse(_message.Message):
     PAGINATED_FIELD_NUMBER: _ClassVar[int]
     code: int
     success: bool
-    data: _containers.RepeatedCompositeFieldContainer[_common_pb2.Telemetry]
+    data: _containers.RepeatedCompositeFieldContainer[TelemetryRecord]
     error: _common_pb2.Error
     paginated: _common_pb2.Paginated
-    def __init__(self, code: _Optional[int] = ..., success: bool = ..., data: _Optional[_Iterable[_Union[_common_pb2.Telemetry, _Mapping]]] = ..., error: _Optional[_Union[_common_pb2.Error, _Mapping]] = ..., paginated: _Optional[_Union[_common_pb2.Paginated, _Mapping]] = ...) -> None: ...
+    def __init__(self, code: _Optional[int] = ..., success: bool = ..., data: _Optional[_Iterable[_Union[TelemetryRecord, _Mapping]]] = ..., error: _Optional[_Union[_common_pb2.Error, _Mapping]] = ..., paginated: _Optional[_Union[_common_pb2.Paginated, _Mapping]] = ...) -> None: ...
 
 class GetAllAssistantResponse(_message.Message):
     __slots__ = ("code", "success", "data", "error", "paginated")

rapida/clients/protos/assistant_api_pb2.py

@@ -11,7 +11,7 @@
 import rapida.clients.protos.assistant_webhook_pb2 as assistant__webhook__pb2
 import rapida.clients.protos.common_pb2 as common__pb2
 
-GRPC_GENERATED_VERSION = '1.78.1'
+GRPC_GENERATED_VERSION = '1.78.0'
 GRPC_VERSION = grpc.__version__
 _version_not_supported = False