Commit 010360f2 authored by Kantz's avatar Kantz
Browse files

kleine restrukturierung

parent e8105cb5
...@@ -15,6 +15,12 @@ def _filter_kwargs(func, kwargs: dict) -> dict: ...@@ -15,6 +15,12 @@ def _filter_kwargs(func, kwargs: dict) -> dict:
return kwargs return kwargs
return {key: value for key, value in kwargs.items() if key in signature.parameters} return {key: value for key, value in kwargs.items() if key in signature.parameters}
def _extract_message(response) -> object:
if isinstance(response, dict) and "message" in response:
return response["message"]
if hasattr(response, "message"):
return response.message
return {}
def _chat_openai(messages: list[dict]) -> dict: def _chat_openai(messages: list[dict]) -> dict:
settings = config.get_openai_chat_settings() settings = config.get_openai_chat_settings()
...@@ -32,6 +38,15 @@ def _chat_openai(messages: list[dict]) -> dict: ...@@ -32,6 +38,15 @@ def _chat_openai(messages: list[dict]) -> dict:
return {"raw": response, "message": message} return {"raw": response, "message": message}
def get_message_content(result: dict | object) -> str:
message = result.get("message") if isinstance(result, dict) else result
if isinstance(message, dict):
return message.get("content", "") or ""
if hasattr(message, "content"):
return getattr(message, "content") or ""
return ""
def chat( def chat(
messages: list[dict], messages: list[dict],
tools: list[Callable[..., Any]] | None = None, tools: list[Callable[..., Any]] | None = None,
...@@ -64,23 +79,6 @@ def chat( ...@@ -64,23 +79,6 @@ def chat(
return {"raw": response, "message": _extract_message(response)} return {"raw": response, "message": _extract_message(response)}
def _extract_message(response) -> object:
if isinstance(response, dict) and "message" in response:
return response["message"]
if hasattr(response, "message"):
return response.message
return {}
def get_message_content(result: dict | object) -> str:
message = result.get("message") if isinstance(result, dict) else result
if isinstance(message, dict):
return message.get("content", "") or ""
if hasattr(message, "content"):
return getattr(message, "content") or ""
return ""
def chat_with_tools( def chat_with_tools(
messages: list[dict], messages: list[dict],
tools: list[Callable[..., Any]], tools: list[Callable[..., Any]],
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment