[model-gateway] Improve logging in data_connector module (#15495)
This commit is contained in:
@@ -1,8 +1,6 @@
|
|||||||
// factory.rs
|
// factory.rs
|
||||||
//
|
//
|
||||||
// Factory function to create storage backends based on configuration.
|
// Factory function to create storage backends based on configuration.
|
||||||
// This centralizes storage initialization logic and fixes the bug where
|
|
||||||
// conversation_item_storage was missing/incorrect in server.rs.
|
|
||||||
|
|
||||||
use std::sync::Arc;
|
use std::sync::Arc;
|
||||||
|
|
||||||
|
|||||||
@@ -332,9 +332,11 @@ impl ResponseStorage for MemoryResponseStorage {
|
|||||||
.push(response_id.clone());
|
.push(response_id.clone());
|
||||||
}
|
}
|
||||||
|
|
||||||
// Store the response
|
|
||||||
store.responses.insert(response_id.clone(), response);
|
store.responses.insert(response_id.clone(), response);
|
||||||
tracing::info!("memory_store_size" = store.responses.len());
|
tracing::debug!(
|
||||||
|
memory_store_size = store.responses.len(),
|
||||||
|
"Response stored in memory"
|
||||||
|
);
|
||||||
|
|
||||||
Ok(response_id)
|
Ok(response_id)
|
||||||
}
|
}
|
||||||
@@ -345,7 +347,7 @@ impl ResponseStorage for MemoryResponseStorage {
|
|||||||
) -> ResponseResult<Option<StoredResponse>> {
|
) -> ResponseResult<Option<StoredResponse>> {
|
||||||
let store = self.store.read();
|
let store = self.store.read();
|
||||||
let result = store.responses.get(response_id).cloned();
|
let result = store.responses.get(response_id).cloned();
|
||||||
tracing::info!("memory_get_response" = %response_id.0, found = result.is_some());
|
tracing::debug!(response_id = %response_id.0, found = result.is_some(), "Memory response lookup");
|
||||||
Ok(result)
|
Ok(result)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -13,6 +13,7 @@ use chrono::{DateTime, Utc};
|
|||||||
use deadpool_postgres::{Manager, ManagerConfig, Pool, RecyclingMethod};
|
use deadpool_postgres::{Manager, ManagerConfig, Pool, RecyclingMethod};
|
||||||
use serde_json::Value;
|
use serde_json::Value;
|
||||||
use tokio_postgres::{NoTls, Row};
|
use tokio_postgres::{NoTls, Row};
|
||||||
|
use tracing;
|
||||||
|
|
||||||
use crate::{
|
use crate::{
|
||||||
config::PostgresConfig,
|
config::PostgresConfig,
|
||||||
@@ -588,7 +589,7 @@ impl ResponseStorage for PostgresResponseStorage {
|
|||||||
&conversation_id,
|
&conversation_id,
|
||||||
&json_raw_response,
|
&json_raw_response,
|
||||||
]).await.unwrap();
|
]).await.unwrap();
|
||||||
println!("INSERT INTO responses VALUES {insert_count}");
|
tracing::debug!(rows_affected = insert_count, "Response stored in Postgres");
|
||||||
Ok(response_id)
|
Ok(response_id)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user