test streaming logs to langsmith

This commit is contained in:
Ishaan Jaff 2024-07-17 16:25:32 -07:00
parent f21a92cadc
commit f945e7009b
2 changed files with 34 additions and 4 deletions

View file

@ -119,9 +119,13 @@ class LangsmithLogger(CustomLogger):
"session_name": project_name,
"start_time": start_time,
"end_time": end_time,
"id": run_id,
}
if run_id:
data["id"] = run_id
verbose_logger.debug("Langsmith Logging data on langsmith: %s", data)
return data
async def async_log_success_event(self, kwargs, response_obj, start_time, end_time):
@ -172,7 +176,7 @@ class LangsmithLogger(CustomLogger):
)
if response.status_code >= 300:
verbose_logger.error(f"Error: {response.status_code}")
verbose_logger.error(f"Error: {response.status_code} - {response.text}")
else:
verbose_logger.debug("Run successfully created")
verbose_logger.debug(