Spaces:
Sleeping
Sleeping
File size: 1,163 Bytes
6a422c8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 |
class CustomOutputWrapper:
@staticmethod
def handle_message_chat_completion(response):
message = {'content': ''}
try:
# handle dashscope response
if 'choices' not in response:
response = response['output']
return response['choices'][0]['message']
except Exception as e:
print(f'input: {response}, original error: {str(e)}')
return message
@staticmethod
def handle_message_chat_completion_chunk(response):
message = {}
try:
return response['choices'][0]['delta']['content']
except Exception as e:
print(f'input: {response}, original error: {str(e)}')
return message
@staticmethod
def handle_message_text_completion(response):
message = ''
try:
message = response['output']['text']
return message
except Exception as e:
print(f'input: {response}, original error: {str(e)}')
return message
DEFAULT_MESSAGE = {
'role': 'user',
'content': 'No entry from user - please suggest something to enter'
}
|