mirror of
https://github.com/trustgraph-ai/trustgraph.git
synced 2026-06-11 07:45:13 +02:00
03cc5ac80f
Replace singleton LibrarianClient with per-flow instances via the new LibrarianSpec, giving each flow its own librarian tied to the workspace-scoped request/response queues from the blueprint. Move all workspace-scoped services (config, flow, librarian, knowledge) from a single base-queue response producer to per-workspace response producers created alongside the existing per-workspace request consumers. Update the gateway dispatcher and bootstrapper flow client to subscribe to the matching workspace-scoped response queues. Fix WorkspaceInit to register workspaces through the IAM create-workspace API so they appear in __workspaces__ and are visible to the gateway. Simplify the bootstrapper gate to only check config-svc reachability. Updated tests accordingly.
47 lines
2.3 KiB
Python
47 lines
2.3 KiB
Python
|
|
from . pubsub import get_pubsub, add_pubsub_args
|
|
from . async_processor import AsyncProcessor
|
|
from . consumer import Consumer
|
|
from . producer import Producer
|
|
from . publisher import Publisher
|
|
from . subscriber import Subscriber
|
|
from . metrics import ProcessorMetrics, ConsumerMetrics, ProducerMetrics, SubscriberMetrics
|
|
from . logging import add_logging_args, setup_logging
|
|
from . workspace_processor import WorkspaceProcessor
|
|
from . flow_processor import FlowProcessor
|
|
from . consumer_spec import ConsumerSpec
|
|
from . parameter_spec import ParameterSpec
|
|
from . producer_spec import ProducerSpec
|
|
from . subscriber_spec import SubscriberSpec
|
|
from . request_response_spec import RequestResponseSpec
|
|
from . llm_service import LlmService, LlmResult, LlmChunk
|
|
from . librarian_client import LibrarianClient
|
|
from . librarian_spec import LibrarianSpec
|
|
from . chunking_service import ChunkingService
|
|
from . embeddings_service import EmbeddingsService
|
|
from . embeddings_client import EmbeddingsClientSpec
|
|
from . text_completion_client import (
|
|
TextCompletionClientSpec, TextCompletionClient, TextCompletionResult,
|
|
)
|
|
from . prompt_client import PromptClientSpec, PromptClient, PromptResult
|
|
from . config_client import ConfigClientSpec, ConfigClient
|
|
from . triples_store_service import TriplesStoreService
|
|
from . graph_embeddings_store_service import GraphEmbeddingsStoreService
|
|
from . document_embeddings_store_service import DocumentEmbeddingsStoreService
|
|
from . triples_query_service import TriplesQueryService
|
|
from . graph_embeddings_query_service import GraphEmbeddingsQueryService
|
|
from . document_embeddings_query_service import DocumentEmbeddingsQueryService
|
|
from . graph_embeddings_client import GraphEmbeddingsClientSpec
|
|
from . triples_client import TriplesClientSpec
|
|
from . document_embeddings_client import DocumentEmbeddingsClientSpec
|
|
from . agent_service import AgentService
|
|
from . graph_rag_client import GraphRagClientSpec
|
|
from . tool_service import ToolService
|
|
from . tool_client import ToolClientSpec
|
|
from . dynamic_tool_service import DynamicToolService
|
|
from . tool_service_client import ToolServiceClientSpec
|
|
from . agent_client import AgentClientSpec
|
|
from . structured_query_client import StructuredQueryClientSpec
|
|
from . row_embeddings_query_client import RowEmbeddingsQueryClientSpec
|
|
from . collection_config_handler import CollectionConfigHandler
|
|
|