mirror of
https://github.com/smkrv/ha-text-ai.git
synced 2026-07-28 01:54:00 +08:00
Release v2.0.0
This commit is contained in:
@@ -328,23 +328,41 @@ class HATextAICoordinator(DataUpdateCoordinator):
|
|||||||
system_prompt: Optional[str]
|
system_prompt: Optional[str]
|
||||||
) -> Dict[str, Any]:
|
) -> Dict[str, Any]:
|
||||||
"""Make API call to OpenAI."""
|
"""Make API call to OpenAI."""
|
||||||
messages = []
|
try:
|
||||||
if system_prompt:
|
messages = []
|
||||||
messages.append({"role": "system", "content": system_prompt})
|
if system_prompt:
|
||||||
messages.append({"role": "user", "content": question})
|
messages.append({"role": "system", "content": system_prompt})
|
||||||
|
messages.append({"role": "user", "content": question})
|
||||||
|
|
||||||
completion = await self.client.chat.completions.create(
|
_LOGGER.debug("Making API call with parameters: model=%s, messages=%s",
|
||||||
model=model or self.model,
|
model or self.model, messages)
|
||||||
messages=messages,
|
|
||||||
temperature=temperature if temperature is not None else self.temperature,
|
|
||||||
max_tokens=max_tokens if max_tokens is not None else self.max_tokens,
|
|
||||||
)
|
|
||||||
|
|
||||||
return {
|
completion = await self.client.chat.completions.create(
|
||||||
"response": completion.choices[0].message.content,
|
model=model or self.model,
|
||||||
"model": completion.model,
|
messages=messages,
|
||||||
"tokens": completion.usage.total_tokens if hasattr(completion, 'usage') else 0
|
temperature=temperature if temperature is not None else self.temperature,
|
||||||
}
|
max_tokens=max_tokens if max_tokens is not None else self.max_tokens,
|
||||||
|
)
|
||||||
|
|
||||||
|
_LOGGER.debug("Received API response: %s", completion)
|
||||||
|
|
||||||
|
if not completion or not hasattr(completion, 'choices') or not completion.choices:
|
||||||
|
raise ValueError("Invalid or empty response from API")
|
||||||
|
|
||||||
|
# Безопасное извлечение сообщения
|
||||||
|
message = completion.choices[0].message
|
||||||
|
if not message or not hasattr(message, 'content'):
|
||||||
|
raise ValueError("No content in API response message")
|
||||||
|
|
||||||
|
return {
|
||||||
|
"response": message.content,
|
||||||
|
"model": getattr(completion, 'model', model or self.model),
|
||||||
|
"tokens": completion.usage.total_tokens if hasattr(completion, 'usage') else 0
|
||||||
|
}
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
_LOGGER.error("OpenAI API call error: %s", str(e))
|
||||||
|
raise
|
||||||
|
|
||||||
async def async_ask_question(
|
async def async_ask_question(
|
||||||
self,
|
self,
|
||||||
|
|||||||
Reference in New Issue
Block a user