Skip to content

Commit 59b4176

Browse files
author
Austin Zielman
committed
Version 1.3.1
1 parent f2440a6 commit 59b4176

File tree

8 files changed

+30
-10
lines changed

8 files changed

+30
-10
lines changed

abacusai/api_class/abstract.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,10 +2,17 @@
22
import datetime
33
import inspect
44
import re
5+
import sys
56
from abc import ABC
67
from copy import deepcopy
78
from textwrap import dedent
8-
from typing import Any, Callable, get_origin, get_type_hints
9+
from typing import Any, Callable, get_type_hints
10+
11+
12+
if sys.version_info >= (3, 8):
13+
from typing import get_origin
14+
else:
15+
from typing_inspect import get_origin
916

1017
from .enums import ApiEnum
1118

abacusai/deployment_conversation_event.py

Lines changed: 9 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -29,9 +29,11 @@ class DeploymentConversationEvent(AbstractApiClass):
2929
segments (list): The segments of the message.
3030
streamedData (str): Aggregated streamed messages from the agent.
3131
streamedSectionData (str): Aggregated streamed section outputs from the agent in a list.
32+
llmDisplayName (str): The display name of the LLM model used to generate the response. Only used for system-created bots.
33+
llmBotIcon (str): The icon location of the LLM model used to generate the response. Only used for system-created bots.
3234
"""
3335

34-
def __init__(self, client, role=None, text=None, timestamp=None, messageIndex=None, regenerateAttempt=None, modelVersion=None, searchResults=None, isUseful=None, feedback=None, feedbackType=None, docInfos=None, keywordArguments=None, inputParams=None, attachments=None, responseVersion=None, agentWorkflowNodeId=None, chatType=None, agentResponse=None, error=None, segments=None, streamedData=None, streamedSectionData=None):
36+
def __init__(self, client, role=None, text=None, timestamp=None, messageIndex=None, regenerateAttempt=None, modelVersion=None, searchResults=None, isUseful=None, feedback=None, feedbackType=None, docInfos=None, keywordArguments=None, inputParams=None, attachments=None, responseVersion=None, agentWorkflowNodeId=None, chatType=None, agentResponse=None, error=None, segments=None, streamedData=None, streamedSectionData=None, llmDisplayName=None, llmBotIcon=None):
3537
super().__init__(client, None)
3638
self.role = role
3739
self.text = text
@@ -55,11 +57,13 @@ def __init__(self, client, role=None, text=None, timestamp=None, messageIndex=No
5557
self.segments = segments
5658
self.streamed_data = streamedData
5759
self.streamed_section_data = streamedSectionData
60+
self.llm_display_name = llmDisplayName
61+
self.llm_bot_icon = llmBotIcon
5862
self.deprecated_keys = {}
5963

6064
def __repr__(self):
61-
repr_dict = {f'role': repr(self.role), f'text': repr(self.text), f'timestamp': repr(self.timestamp), f'message_index': repr(self.message_index), f'regenerate_attempt': repr(self.regenerate_attempt), f'model_version': repr(self.model_version), f'search_results': repr(self.search_results), f'is_useful': repr(self.is_useful), f'feedback': repr(self.feedback), f'feedback_type': repr(self.feedback_type), f'doc_infos': repr(self.doc_infos), f'keyword_arguments': repr(
62-
self.keyword_arguments), f'input_params': repr(self.input_params), f'attachments': repr(self.attachments), f'response_version': repr(self.response_version), f'agent_workflow_node_id': repr(self.agent_workflow_node_id), f'chat_type': repr(self.chat_type), f'agent_response': repr(self.agent_response), f'error': repr(self.error), f'segments': repr(self.segments), f'streamed_data': repr(self.streamed_data), f'streamed_section_data': repr(self.streamed_section_data)}
65+
repr_dict = {f'role': repr(self.role), f'text': repr(self.text), f'timestamp': repr(self.timestamp), f'message_index': repr(self.message_index), f'regenerate_attempt': repr(self.regenerate_attempt), f'model_version': repr(self.model_version), f'search_results': repr(self.search_results), f'is_useful': repr(self.is_useful), f'feedback': repr(self.feedback), f'feedback_type': repr(self.feedback_type), f'doc_infos': repr(self.doc_infos), f'keyword_arguments': repr(self.keyword_arguments), f'input_params': repr(
66+
self.input_params), f'attachments': repr(self.attachments), f'response_version': repr(self.response_version), f'agent_workflow_node_id': repr(self.agent_workflow_node_id), f'chat_type': repr(self.chat_type), f'agent_response': repr(self.agent_response), f'error': repr(self.error), f'segments': repr(self.segments), f'streamed_data': repr(self.streamed_data), f'streamed_section_data': repr(self.streamed_section_data), f'llm_display_name': repr(self.llm_display_name), f'llm_bot_icon': repr(self.llm_bot_icon)}
6367
class_name = "DeploymentConversationEvent"
6468
repr_str = ',\n '.join([f'{key}={value}' for key, value in repr_dict.items(
6569
) if getattr(self, key, None) is not None and key not in self.deprecated_keys])
@@ -72,6 +76,6 @@ def to_dict(self):
7276
Returns:
7377
dict: The dict value representation of the class parameters
7478
"""
75-
resp = {'role': self.role, 'text': self.text, 'timestamp': self.timestamp, 'message_index': self.message_index, 'regenerate_attempt': self.regenerate_attempt, 'model_version': self.model_version, 'search_results': self.search_results, 'is_useful': self.is_useful, 'feedback': self.feedback, 'feedback_type': self.feedback_type, 'doc_infos': self.doc_infos, 'keyword_arguments': self.keyword_arguments,
76-
'input_params': self.input_params, 'attachments': self.attachments, 'response_version': self.response_version, 'agent_workflow_node_id': self.agent_workflow_node_id, 'chat_type': self.chat_type, 'agent_response': self.agent_response, 'error': self.error, 'segments': self.segments, 'streamed_data': self.streamed_data, 'streamed_section_data': self.streamed_section_data}
79+
resp = {'role': self.role, 'text': self.text, 'timestamp': self.timestamp, 'message_index': self.message_index, 'regenerate_attempt': self.regenerate_attempt, 'model_version': self.model_version, 'search_results': self.search_results, 'is_useful': self.is_useful, 'feedback': self.feedback, 'feedback_type': self.feedback_type, 'doc_infos': self.doc_infos, 'keyword_arguments': self.keyword_arguments, 'input_params': self.input_params,
80+
'attachments': self.attachments, 'response_version': self.response_version, 'agent_workflow_node_id': self.agent_workflow_node_id, 'chat_type': self.chat_type, 'agent_response': self.agent_response, 'error': self.error, 'segments': self.segments, 'streamed_data': self.streamed_data, 'streamed_section_data': self.streamed_section_data, 'llm_display_name': self.llm_display_name, 'llm_bot_icon': self.llm_bot_icon}
7781
return {key: value for key, value in resp.items() if value is not None and key not in self.deprecated_keys}

abacusai/requirements.txt

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,4 @@
11
requests >= 2.22.0
22
packaging >= 20.1
3+
typing_inspect
4+
python_version < "3.8"

docs/_sources/autoapi/abacusai/deployment_conversation_event/index.rst.txt

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ Classes
1717

1818

1919

20-
.. py:class:: DeploymentConversationEvent(client, role=None, text=None, timestamp=None, messageIndex=None, regenerateAttempt=None, modelVersion=None, searchResults=None, isUseful=None, feedback=None, feedbackType=None, docInfos=None, keywordArguments=None, inputParams=None, attachments=None, responseVersion=None, agentWorkflowNodeId=None, chatType=None, agentResponse=None, error=None, segments=None, streamedData=None, streamedSectionData=None)
20+
.. py:class:: DeploymentConversationEvent(client, role=None, text=None, timestamp=None, messageIndex=None, regenerateAttempt=None, modelVersion=None, searchResults=None, isUseful=None, feedback=None, feedbackType=None, docInfos=None, keywordArguments=None, inputParams=None, attachments=None, responseVersion=None, agentWorkflowNodeId=None, chatType=None, agentResponse=None, error=None, segments=None, streamedData=None, streamedSectionData=None, llmDisplayName=None, llmBotIcon=None)
2121
2222
2323
Bases: :py:obj:`abacusai.return_class.AbstractApiClass`
@@ -70,6 +70,10 @@ Classes
7070
:type streamedData: str
7171
:param streamedSectionData: Aggregated streamed section outputs from the agent in a list.
7272
:type streamedSectionData: str
73+
:param llmDisplayName: The display name of the LLM model used to generate the response. Only used for system-created bots.
74+
:type llmDisplayName: str
75+
:param llmBotIcon: The icon location of the LLM model used to generate the response. Only used for system-created bots.
76+
:type llmBotIcon: str
7377

7478
.. py:method:: __repr__()
7579

0 commit comments

Comments
 (0)