first commit
This commit is contained in:
@@ -0,0 +1,37 @@
|
||||
"""
|
||||
Server state management module.
|
||||
"""
|
||||
|
||||
from typing import Optional, Any
|
||||
from context.conversation_manager import ConversationManager, get_conversation_manager
|
||||
from agent import get_agent_registry
|
||||
|
||||
|
||||
class ServerState:
|
||||
"""Server state management class, replacing global variables."""
|
||||
|
||||
def __init__(self):
|
||||
self.agent_registry: Optional[Any] = None
|
||||
self.conversation_manager: Optional[ConversationManager] = None
|
||||
|
||||
def initialize(self) -> None:
|
||||
"""Initialize server state."""
|
||||
self.agent_registry = get_agent_registry()
|
||||
self.conversation_manager = get_conversation_manager()
|
||||
|
||||
# Create the default Agent instance.
|
||||
self.agent_registry.get_or_create_agent(
|
||||
"cadagent",
|
||||
name="CADAgent Web Service",
|
||||
description="Professional CAD modeling assistant with web API",
|
||||
max_history_length=4, # DEFAULT_MAX_HISTORY_LENGTH
|
||||
)
|
||||
|
||||
|
||||
# Global server state.
|
||||
server_state = ServerState()
|
||||
|
||||
|
||||
def get_server_state() -> ServerState:
|
||||
"""Get the server state dependency."""
|
||||
return server_state
|
||||
Reference in New Issue
Block a user