Merge pull request #2 from rapidaai/feature-agentkit

feat: added agentkit

Author: iamprashant

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "rapida-python"
-version = "0.1.24"
+version = "0.1.25"
 description = "RapidaAI SDK to integrate rapida.ai APIs"
 readme = "README.md"
 authors = [{name = "RapidaAI", email = "code@rapida.ai"}]
@@ -45,12 +45,12 @@ Source = "https://github.com/rapidaai/rapida-python"
 [tool.setuptools]
 packages = [
     "rapida",
+    "rapida.agentkit",
     "rapida.clients",
     "rapida.clients.protos",
     "rapida.configs",
     "rapida.connections",
     "rapida.utils"
-    # Add any other packages that should be included
 ]
 
 [tool.setuptools.package-data]

rapida/__init__.py

@@ -109,7 +109,6 @@
     Tag,
     Organization,
     Metric,
-    Content,
     Message as ProtoMessage,
     ToolCall,
     FunctionCall,
@@ -123,14 +122,7 @@
     GetAllAssistantConversationResponse,
     GetAllConversationMessageRequest,
     GetAllConversationMessageResponse,
-    AssistantConversationMessageTextContent,
-    AssistantConversationMessageAudioContent,
     AssistantDefinition,
-    AssistantConversationConfiguration,
-    AssistantConversationInterruption,
-    AssistantConversationUserMessage,
-    AssistantConversationAction,
-    AssistantConversationAssistantMessage,
 )
 
 # Now import other protos that may depend on common_pb2
@@ -162,8 +154,6 @@
     GetAllOrganizationCredentialRequest,
 )
 from rapida.clients.protos.talk_api_pb2 import (
-    AssistantMessagingRequest,
-    AssistantMessagingResponse,
     CreateMessageMetricRequest,
     CreateMessageMetricResponse,
     CreateConversationMetricRequest,
@@ -172,6 +162,17 @@
     CreateBulkPhoneCallResponse,
     CreatePhoneCallRequest,
     CreatePhoneCallResponse,
+    AssistantTalkInput,
+    AssistantTalkOutput,
+    TalkInput,
+    TalkOutput,
+    ConversationAssistantMessage,
+    ConversationConfiguration,
+    ConversationDirective,
+    ConversationInterruption,
+    ConversationUserMessage,
+    ConversationToolCall,
+    ConversationToolResult
 )
 from rapida.clients.protos.assistant_analysis_pb2 import (
     AssistantAnalysis,
@@ -383,6 +384,15 @@
     add_AgentKitServicer_to_server,
     )
 
+# Agent Kit classes
+from rapida.agentkit import (
+    AgentKitAgent,
+    AgentKitServer,
+    SSLConfig,
+    AuthConfig,
+    AuthorizationInterceptor,
+)
+
 __all__ = [
     "Any",
     "RapidaSource",
@@ -468,12 +478,6 @@
     "GetCredentialResponse",
     "GetAllOrganizationCredentialRequest",
     "AssistantDefinition",
-    "AssistantMessagingRequest",
-    "AssistantConversationConfiguration",
-    "AssistantConversationInterruption",
-    "AssistantConversationUserMessage",
-    "AssistantConversationAssistantMessage",
-    "AssistantMessagingResponse",
     "CreateMessageMetricRequest",
     "CreateMessageMetricResponse",
     "CreateConversationMetricRequest",
@@ -561,7 +565,6 @@
     "Tag",
     "Organization",
     "Metric",
-    "Content",
     "ProtoMessage",
     "ToolCall",
     "FunctionCall",
@@ -689,12 +692,26 @@
     "UpdateAssistantDetailRequest",
     "GetAssistantConversationRequest",
     "GetAssistantConversationResponse",
-    "AssistantConversationMessageTextContent",
-    "AssistantConversationMessageAudioContent",
     "AgentKitStub",
     "AgentKit",
     "AgentKitServicer",
     "TalkServiceServicer",
     "add_AgentKitServicer_to_server",
-    "AssistantConversationAction",
+    # Agent Kit
+    "AgentKitAgent",
+    "AgentKitServer",
+    "SSLConfig",
+    "AuthConfig",
+    "AuthorizationInterceptor",
+    "AssistantTalkInput",
+    "AssistantTalkOutput",
+    "TalkInput",
+    "TalkOutput",
+    "ConversationAssistantMessage",
+    "ConversationConfiguration",
+    "ConversationDirective",
+    "ConversationInterruption",
+    "ConversationUserMessage",
+    "ConversationToolCall",
+    "ConversationToolResult"
 ]