@@ -1116,50 +1116,59 @@ async def test_decrypt_file_data_read_chunks(self):
1116
1116
async def test_converse_alpha1_basic (self ):
1117
1117
dapr = DaprGrpcClientAsync (f'{ self .scheme } localhost:{ self .grpc_port } ' )
1118
1118
1119
- inputs = [ConversationInput (message = "Hello" , role = "user" ),
1120
- ConversationInput (message = "How are you?" , role = "user" )]
1119
+ inputs = [
1120
+ ConversationInput (message = 'Hello' , role = 'user' ),
1121
+ ConversationInput (message = 'How are you?' , role = 'user' ),
1122
+ ]
1121
1123
1122
- response = await dapr .converse_alpha1 (name = " test-llm" , inputs = inputs )
1124
+ response = await dapr .converse_alpha1 (name = ' test-llm' , inputs = inputs )
1123
1125
1124
1126
# Check response structure
1125
1127
self .assertIsNotNone (response )
1126
1128
self .assertEqual (len (response .outputs ), 2 )
1127
- self .assertEqual (response .outputs [0 ].result , " Response to: Hello" )
1128
- self .assertEqual (response .outputs [1 ].result , " Response to: How are you?" )
1129
+ self .assertEqual (response .outputs [0 ].result , ' Response to: Hello' )
1130
+ self .assertEqual (response .outputs [1 ].result , ' Response to: How are you?' )
1129
1131
await dapr .close ()
1130
1132
1131
1133
async def test_converse_alpha1_with_options (self ):
1132
1134
dapr = DaprGrpcClientAsync (f'{ self .scheme } localhost:{ self .grpc_port } ' )
1133
1135
1134
- inputs = [ConversationInput (message = " Hello" , role = " user" , scrub_pii = True )]
1136
+ inputs = [ConversationInput (message = ' Hello' , role = ' user' , scrub_pii = True )]
1135
1137
1136
- response = await dapr .converse_alpha1 (name = "test-llm" , inputs = inputs , context_id = "chat-123" ,
1137
- temperature = 0.7 , scrub_pii = True , metadata = {"key" : "value" })
1138
+ response = await dapr .converse_alpha1 (
1139
+ name = 'test-llm' ,
1140
+ inputs = inputs ,
1141
+ context_id = 'chat-123' ,
1142
+ temperature = 0.7 ,
1143
+ scrub_pii = True ,
1144
+ metadata = {'key' : 'value' },
1145
+ )
1138
1146
1139
1147
self .assertIsNotNone (response )
1140
1148
self .assertEqual (len (response .outputs ), 1 )
1141
- self .assertEqual (response .outputs [0 ].result , " Response to: Hello" )
1149
+ self .assertEqual (response .outputs [0 ].result , ' Response to: Hello' )
1142
1150
await dapr .close ()
1143
1151
1144
1152
async def test_converse_alpha1_error_handling (self ):
1145
1153
dapr = DaprGrpcClientAsync (f'{ self .scheme } localhost:{ self .grpc_port } ' )
1146
1154
1147
1155
# Setup server to raise an exception
1148
1156
self ._fake_dapr_server .raise_exception_on_next_call (
1149
- status_pb2 .Status (code = code_pb2 .INVALID_ARGUMENT , message = "Invalid argument" ))
1157
+ status_pb2 .Status (code = code_pb2 .INVALID_ARGUMENT , message = 'Invalid argument' )
1158
+ )
1150
1159
1151
- inputs = [ConversationInput (message = " Hello" , role = " user" )]
1160
+ inputs = [ConversationInput (message = ' Hello' , role = ' user' )]
1152
1161
1153
1162
with self .assertRaises (DaprInternalError ) as context :
1154
- await dapr .converse_alpha1 (name = " test-llm" , inputs = inputs )
1155
- self .assertTrue (" Invalid argument" in str (context .exception ))
1163
+ await dapr .converse_alpha1 (name = ' test-llm' , inputs = inputs )
1164
+ self .assertTrue (' Invalid argument' in str (context .exception ))
1156
1165
await dapr .close ()
1157
1166
1158
1167
async def test_converse_alpha1_empty_inputs (self ):
1159
1168
dapr = DaprGrpcClientAsync (f'{ self .scheme } localhost:{ self .grpc_port } ' )
1160
1169
1161
1170
# Test with empty inputs list
1162
- response = await dapr .converse_alpha1 (name = " test-llm" , inputs = [])
1171
+ response = await dapr .converse_alpha1 (name = ' test-llm' , inputs = [])
1163
1172
1164
1173
self .assertIsNotNone (response )
1165
1174
self .assertEqual (len (response .outputs ), 0 )
0 commit comments