gmakstutis/openrag
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

logging

Browse source
This commit is contained in:
phact 2025-09-08 16:07:06 -04:00
parent 982726d904
commit 73f039afb5
1 changed files with 30 additions and 23 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

53
src/agent.py
View file

@ -189,35 +189,42 @@ async def async_response(
previous_response_id: str = None, previous_response_id: str = None,
log_prefix: str = "response", log_prefix: str = "response",
): ):
logger.info("User prompt received", prompt=prompt) try:
logger.info("User prompt received", prompt=prompt)
# Build request parameters # Build request parameters
request_params = { request_params = {
"model": model, "model": model,
"input": prompt, "input": prompt,
"stream": False, "stream": False,
"include": ["tool_call.results"], "include": ["tool_call.results"],
} }
if previous_response_id is not None: if previous_response_id is not None:
request_params["previous_response_id"] = previous_response_id request_params["previous_response_id"] = previous_response_id
if extra_headers: if extra_headers:
request_params["extra_headers"] = extra_headers request_params["extra_headers"] = extra_headers
if "x-api-key" not in client.default_headers: if "x-api-key" not in client.default_headers:
if hasattr(client, "api_key") and extra_headers is not None: if hasattr(client, "api_key") and extra_headers is not None:
extra_headers["x-api-key"] = client.api_key extra_headers["x-api-key"] = client.api_key
response = await client.responses.create(**request_params) response = await client.responses.create(**request_params)
response_text = response.output_text response_text = response.output_text
logger.info("Response generated", log_prefix=log_prefix, response=response_text) logger.info("Response generated", log_prefix=log_prefix, response=response_text)
# Extract and store response_id if available # Extract and store response_id if available
response_id = getattr(response, "id", None) or getattr( response_id = getattr(response, "id", None) or getattr(
response, "response_id", None response, "response_id", None
) )
return response_text, response_id, response return response_text, response_id, response
except Exception as e:
logger.error("Exception in non-streaming response", error=str(e))
import traceback
traceback.print_exc()
raise
# Unified streaming function for both chat and langflow # Unified streaming function for both chat and langflow