Skip to content

Commit f62b761

Browse files
committed
feat: updated api for assistant
1 parent 482e2ca commit f62b761

3 files changed

Lines changed: 38 additions & 3 deletions

File tree

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
44

55
[project]
66
name = "rapida-python"
7-
version = "0.1.15"
7+
version = "0.1.16"
88
description = "RapidaAI SDK to integrate rapida.ai APIs"
99
readme = "README.md"
1010
authors = [{name = "RapidaAI", email = "code@rapida.ai"}]

rapida/__init__.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -213,6 +213,8 @@
213213
get_all_assistant_tool,
214214
get_assistant_webhook_log,
215215
get_all_assistant_webhook_log,
216+
get_assistant_analysis,
217+
get_all_assistant_analysis,
216218
)
217219

218220
from rapida.clients.endpoint import (
@@ -534,8 +536,6 @@
534536
GetAllMessageResponse,
535537
UpdateAssistantVersionRequest,
536538
UpdateAssistantDetailRequest,
537-
GetAllAssistantUserConversationRequest,
538-
GetAllAssistantUserConversationResponse,
539539
GetAssistantConversationRequest,
540540
GetAssistantConversationResponse,
541541
)
@@ -599,6 +599,8 @@
599599
"get_endpoint_log",
600600
"get_all_endpoint_log",
601601
"invoke",
602+
"get_assistant_analysis",
603+
"get_all_assistant_analysis",
602604
# "IndexKnowledgeDocument",
603605
# "GetAllDeployment",
604606
# "Invoke",

rapida/clients/assistant.py

Lines changed: 33 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,13 @@
2020
GetAssistantWebhookRequest,
2121
GetAssistantWebhookResponse,
2222
)
23+
24+
from rapida.clients.protos.assistant_analysis_pb2 import (
25+
GetAllAssistantAnalysisRequest,
26+
GetAssistantAnalysisRequest,
27+
GetAssistantAnalysisResponse,
28+
GetAllAssistantAnalysisResponse
29+
)
2330
from rapida.clients.protos.assistant_tool_pb2 import (
2431
GetAllAssistantToolRequest,
2532
GetAllAssistantToolResponse,
@@ -167,6 +174,32 @@ def get_all_assistant_tool(
167174
)
168175

169176

177+
def get_assistant_analysis(
178+
client_cfg: ConnectionConfig,
179+
request: GetAssistantAnalysisRequest,
180+
auth: Union[UserAuthInfo, ClientAuthInfo, None] = None,
181+
) -> GetAssistantAnalysisResponse:
182+
if auth is None:
183+
auth = client_cfg.auth
184+
return client_cfg.assistant_client.GetAssistantAnalysis(
185+
request,
186+
metadata=auth,
187+
)
188+
189+
190+
def get_all_assistant_analysis(
191+
client_cfg: ConnectionConfig,
192+
request: GetAllAssistantAnalysisRequest,
193+
auth: Union[UserAuthInfo, ClientAuthInfo, None] = None,
194+
) -> GetAllAssistantAnalysisResponse:
195+
if auth is None:
196+
auth = client_cfg.auth
197+
return client_cfg.assistant_client.GetAllAssistantAnalysis(
198+
request,
199+
metadata=auth,
200+
)
201+
202+
170203
def get_assistant_webhook_log(
171204
client_cfg: ConnectionConfig,
172205
request: GetAssistantWebhookLogRequest,

0 commit comments

Comments
 (0)