MCPcopy Create free account
hub / github.com/Open-Bee/DataStudio / _parse_openai_response

Method _parse_openai_response

datastudio/models/base.py:294–313  ·  view source on GitHub ↗

Parse OpenAI-format API response content. Args: data: Parsed JSON response from OpenAI API. Returns: Extracted content string, with optional reasoning_content formatting.

(self, data: dict)

Source from the content-addressed store, hash-verified

292 return input_msgs, text
293
294 def _parse_openai_response(self, data: dict) -> str:
295 """Parse OpenAI-format API response content.
296
297 Args:
298 data: Parsed JSON response from OpenAI API.
299
300 Returns:
301 Extracted content string, with optional reasoning_content formatting.
302 """
303 message_obj = data.get("choices", [{}])[0].get("message", {})
304 content = message_obj.get("content", self.fail_msg)
305 reasoning_content = message_obj.get("reasoning_content")
306
307 if self.return_dict:
308 return load_str_to_dict(content)
309 if reasoning_content:
310 return (
311 f"<think>\n{reasoning_content.strip()}\n</think>\n\n{content.strip()}"
312 )
313 return content
314
315 def _build_openai_headers(self, api_key: str) -> dict:
316 """Build HTTP headers for OpenAI-compatible API request.

Calls 2

load_str_to_dictFunction · 0.90
getMethod · 0.45