Private
Public Access
refactor(ai_client): _send_deepseek_result() returns Result[str]
This commit is contained in:
+5
-5
@@ -1903,14 +1903,14 @@ def _ensure_deepseek_client() -> None:
|
|||||||
_load_credentials()
|
_load_credentials()
|
||||||
pass
|
pass
|
||||||
|
|
||||||
def _send_deepseek(md_content: str, user_message: str, base_dir: str,
|
def _send_deepseek_result(md_content: str, user_message: str, base_dir: str,
|
||||||
file_items: list[dict[str, Any]] | None = None,
|
file_items: list[dict[str, Any]] | None = None,
|
||||||
discussion_history: str = "",
|
discussion_history: str = "",
|
||||||
stream: bool = False,
|
stream: bool = False,
|
||||||
pre_tool_callback: Optional[Callable[[str, str, Optional[Callable[[str], str]]], Optional[str]]] = None,
|
pre_tool_callback: Optional[Callable[[str, str, Optional[Callable[[str], str]]], Optional[str]]] = None,
|
||||||
qa_callback: Optional[Callable[[str], str]] = None,
|
qa_callback: Optional[Callable[[str], str]] = None,
|
||||||
stream_callback: Optional[Callable[[str], None]] = None,
|
stream_callback: Optional[Callable[[str], None]] = None,
|
||||||
patch_callback: Optional[Callable[[str, str], Optional[str]]] = None) -> str:
|
patch_callback: Optional[Callable[[str, str], Optional[str]]] = None) -> Result[str]:
|
||||||
"""
|
"""
|
||||||
[C: src/ai_server.py:_handle_send]
|
[C: src/ai_server.py:_handle_send]
|
||||||
"""
|
"""
|
||||||
@@ -2007,7 +2007,7 @@ def _send_deepseek(md_content: str, user_message: str, base_dir: str,
|
|||||||
response.raise_for_status()
|
response.raise_for_status()
|
||||||
except requests.exceptions.RequestException as e:
|
except requests.exceptions.RequestException as e:
|
||||||
if monitor.enabled: monitor.end_component("ai_client._send_deepseek")
|
if monitor.enabled: monitor.end_component("ai_client._send_deepseek")
|
||||||
raise _classify_deepseek_error(e) from e
|
return Result(data="", errors=[_classify_deepseek_error(e, source="ai_client.deepseek")])
|
||||||
|
|
||||||
assistant_text = ""
|
assistant_text = ""
|
||||||
tool_calls_raw = []
|
tool_calls_raw = []
|
||||||
@@ -2155,10 +2155,10 @@ def _send_deepseek(md_content: str, user_message: str, base_dir: str,
|
|||||||
|
|
||||||
res = "\n\n".join(all_text_parts) if all_text_parts else "(No text returned)"
|
res = "\n\n".join(all_text_parts) if all_text_parts else "(No text returned)"
|
||||||
if monitor.enabled: monitor.end_component("ai_client._send_deepseek")
|
if monitor.enabled: monitor.end_component("ai_client._send_deepseek")
|
||||||
return res
|
return Result(data=res)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
if monitor.enabled: monitor.end_component("ai_client._send_deepseek")
|
if monitor.enabled: monitor.end_component("ai_client._send_deepseek")
|
||||||
raise _classify_deepseek_error(e) from e
|
return Result(data="", errors=[_classify_deepseek_error(e, source="ai_client.deepseek")])
|
||||||
|
|
||||||
#endregion: DeepSeek Provider
|
#endregion: DeepSeek Provider
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user