Skip to content

Commit ae2dd38

Browse files
committed
Fix pyright/selftest
1 parent edf6db0 commit ae2dd38

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

ansible_ai_connect/ai/api/model_pipelines/http/pipelines.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
import json
1616
import logging
1717
from json import JSONDecodeError
18-
from typing import AsyncGenerator
18+
from typing import AsyncGenerator, Any
1919

2020
import aiohttp
2121
import requests
@@ -181,7 +181,7 @@ def invoke(self, params: ChatBotParameters) -> ChatBotResponse:
181181
system_prompt = params.system_prompt or settings.CHATBOT_DEFAULT_SYSTEM_PROMPT
182182
no_tools = params.no_tools
183183

184-
data = {
184+
data: dict[str, Any] = {
185185
"query": query,
186186
"model": model_id,
187187
"provider": provider,
@@ -286,7 +286,7 @@ async def async_invoke(self, params: StreamingChatBotParameters) -> AsyncGenerat
286286
media_type = params.media_type
287287
no_tools = params.no_tools
288288

289-
data = {
289+
data: dict[str, Any] = {
290290
"query": query,
291291
"model": model_id,
292292
"provider": provider,
@@ -339,9 +339,9 @@ async def async_invoke(self, params: StreamingChatBotParameters) -> AsyncGenerat
339339
logger.error(
340340
"An error received in chat streaming content:"
341341
+ " response="
342-
+ data.get("response")
342+
+ str(data.get("response"))
343343
+ ", cause="
344-
+ data.get("cause")
344+
+ str(data.get("cause"))
345345
)
346346
elif event == "start":
347347
ev.phase = event

0 commit comments

Comments
 (0)