add basic i18n
This commit is contained in:
@@ -7,7 +7,6 @@ use std::thread::JoinHandle;
|
||||
use std::time::Duration;
|
||||
use hound::WavSpec;
|
||||
use log::{debug, error, info, trace};
|
||||
use thiserror::__private18::AsDisplay;
|
||||
use thiserror::Error;
|
||||
|
||||
fn generic_sample_to_i16(sample: impl SizedSample + Into<f64>, format: SampleFormat) -> i16 {
|
||||
|
||||
@@ -103,9 +103,8 @@ impl AgentChat {
|
||||
ollama_client: Ollama,
|
||||
model: String,
|
||||
mcp_clients: HashMap<String, MCPClient>,
|
||||
system_prompt: String,
|
||||
) -> Result<Self, ServiceError> {
|
||||
let history = vec![ChatMessage::system(system_prompt)];
|
||||
let history = vec![];
|
||||
|
||||
let mut servers = HashMap::new();
|
||||
|
||||
@@ -192,7 +191,13 @@ impl AgentChat {
|
||||
{
|
||||
let all_tools: Vec<ToolInfo> = self
|
||||
.get_all_tools()
|
||||
.map(|restricted_tool| restricted_tool.tool_info.clone())
|
||||
.filter_map(|restricted_tool|{
|
||||
if restricted_tool.permission == ToolPermission::Denied { // filters denied tools to save tokens
|
||||
None
|
||||
} else {
|
||||
Some(restricted_tool.tool_info.clone())
|
||||
}
|
||||
})
|
||||
.collect();
|
||||
|
||||
log::debug!("all tools: {all_tools:#?}");
|
||||
@@ -347,7 +352,7 @@ mod tests {
|
||||
).await.unwrap(),
|
||||
)]);
|
||||
|
||||
let mut chat = AgentChat::new(ollama, "lfm2.5-thinking:1.2b".to_string(), mcp_clients, String::new()).await.unwrap();
|
||||
let mut chat = AgentChat::new(ollama, "lfm2.5-thinking:1.2b".to_string(), mcp_clients).await.unwrap();
|
||||
|
||||
let tools: Vec<RestrictedTool> = chat.get_all_tools().cloned().collect();
|
||||
assert_eq!(tools.len(), 1);
|
||||
|
||||
Reference in New Issue
Block a user