Skip to content

Commit b26dfb5

Browse files
committed
Modifications from the server
Signed-off-by: Alex Mikhalev <alex@metacortex.engineer>
1 parent a95fa48 commit b26dfb5

14 files changed

Lines changed: 31 additions & 37 deletions

crates/terraphim_multi_agent/benches/agent_operations.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,9 @@
1-
use criterion::{BenchmarkId, Criterion, black_box, criterion_group, criterion_main};
1+
use criterion::{black_box, criterion_group, criterion_main, BenchmarkId, Criterion};
22
use std::sync::Arc;
33
use tokio::runtime::Runtime;
44

55
use terraphim_multi_agent::{
6-
AgentRegistry, CommandInput, CommandType, test_utils::create_test_agent_simple,
6+
test_utils::create_test_agent_simple, AgentRegistry, CommandInput, CommandType,
77
};
88

99
/// Benchmark agent creation time

crates/terraphim_multi_agent/examples/enhanced_atomic_server_example.rs

Lines changed: 6 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -30,14 +30,12 @@ fn create_atomic_server_agent_role() -> Role {
3030
llm_chat_system_prompt: None,
3131
llm_chat_model: None,
3232
llm_context_window: Some(16000),
33-
haystacks: vec![
34-
Haystack::new(
35-
"http://localhost:9883".to_string(), // Atomic server URL
36-
ServiceType::Atomic,
37-
true, // read-only
38-
)
39-
.with_atomic_secret(Some("your-base64-secret-here".to_string())),
40-
],
33+
haystacks: vec![Haystack::new(
34+
"http://localhost:9883".to_string(), // Atomic server URL
35+
ServiceType::Atomic,
36+
true, // read-only
37+
)
38+
.with_atomic_secret(Some("your-base64-secret-here".to_string()))],
4139
extra: {
4240
let mut extra = AHashMap::new();
4341
// Multi-agent specific configuration

crates/terraphim_multi_agent/examples/knowledge_graph_integration.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99
use std::sync::Arc;
1010
use terraphim_config::Role;
1111
use terraphim_multi_agent::{
12-
CommandInput, CommandType, MultiAgentResult, TerraphimAgent, test_utils::create_test_role,
12+
test_utils::create_test_role, CommandInput, CommandType, MultiAgentResult, TerraphimAgent,
1313
};
1414
use terraphim_persistence::DeviceStorage;
1515

crates/terraphim_multi_agent/examples/multi_agent_coordination.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,8 +10,8 @@ use ahash::AHashMap;
1010
use std::sync::Arc;
1111
use terraphim_config::Role;
1212
use terraphim_multi_agent::{
13-
AgentRegistry, CommandInput, CommandType, MultiAgentResult, TerraphimAgent,
14-
test_utils::create_test_role,
13+
test_utils::create_test_role, AgentRegistry, CommandInput, CommandType, MultiAgentResult,
14+
TerraphimAgent,
1515
};
1616
use terraphim_persistence::DeviceStorage;
1717
use terraphim_types::RelevanceFunction;

crates/terraphim_multi_agent/examples/simple_validation.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
66
use std::sync::Arc;
77
use terraphim_multi_agent::{
8-
CommandInput, CommandType, TerraphimAgent, test_utils::create_test_role,
8+
test_utils::create_test_role, CommandInput, CommandType, TerraphimAgent,
99
};
1010
use terraphim_persistence::DeviceStorage;
1111

crates/terraphim_multi_agent/examples/specialized_agents.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
//! that leverage the generic LLM interface instead of OpenRouter-specific code.
55
66
use terraphim_multi_agent::{
7-
ChatAgent, ChatConfig, SummarizationAgent, SummarizationConfig, SummaryStyle, test_utils,
7+
test_utils, ChatAgent, ChatConfig, SummarizationAgent, SummarizationConfig, SummaryStyle,
88
};
99

1010
#[tokio::main]

crates/terraphim_multi_agent/examples/workflow_patterns_working.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ use ahash::AHashMap;
88
use std::sync::Arc;
99
use terraphim_config::Role;
1010
use terraphim_multi_agent::{
11-
CommandInput, CommandType, MultiAgentResult, TerraphimAgent, test_utils::create_test_role,
11+
test_utils::create_test_role, CommandInput, CommandType, MultiAgentResult, TerraphimAgent,
1212
};
1313
use terraphim_persistence::DeviceStorage;
1414
use terraphim_types::RelevanceFunction;

crates/terraphim_multi_agent/src/agent.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -911,7 +911,7 @@ impl TerraphimAgent {
911911
// TODO: Load from role configuration
912912
// TODO: Load actual automata from role config
913913
// TODO: Load actual automata from role config
914-
use terraphim_automata::{AutocompleteConfig, build_autocomplete_index};
914+
use terraphim_automata::{build_autocomplete_index, AutocompleteConfig};
915915
use terraphim_types::Thesaurus;
916916

917917
let thesaurus = Thesaurus::new("default".to_string());

crates/terraphim_multi_agent/src/genai_llm_client.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,8 +6,8 @@
66
77
use crate::{LlmRequest, LlmResponse, MessageRole, MultiAgentError, MultiAgentResult, TokenUsage};
88
use chrono::Utc;
9-
use genai::Client;
109
use genai::chat::{ChatMessage, ChatOptions, ChatRequest};
10+
use genai::Client;
1111
use uuid::Uuid;
1212

1313
/// Rust-GenAI LLM client that works with multiple providers

crates/terraphim_multi_agent/src/lib.rs

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -122,8 +122,8 @@ pub mod test_utils {
122122
}
123123

124124
/// Create test rolegraph for testing
125-
pub async fn create_test_rolegraph()
126-
-> Result<Arc<terraphim_rolegraph::RoleGraph>, MultiAgentError> {
125+
pub async fn create_test_rolegraph(
126+
) -> Result<Arc<terraphim_rolegraph::RoleGraph>, MultiAgentError> {
127127
// Create a simple test rolegraph with empty thesaurus
128128
use terraphim_types::Thesaurus;
129129
let empty_thesaurus = Thesaurus::new("test_thesaurus".to_string());
@@ -139,8 +139,8 @@ pub mod test_utils {
139139
}
140140

141141
/// Create test automata for testing
142-
pub fn create_test_automata()
143-
-> Result<Arc<terraphim_automata::AutocompleteIndex>, MultiAgentError> {
142+
pub fn create_test_automata(
143+
) -> Result<Arc<terraphim_automata::AutocompleteIndex>, MultiAgentError> {
144144
// Create a simple test automata index with empty thesaurus
145145
use terraphim_types::Thesaurus;
146146
let empty_thesaurus = Thesaurus::new("test_thesaurus".to_string());

0 commit comments

Comments
 (0)