bohr-agent-sdk 0.1.101__py3-none-any.whl → 0.1.103__py3-none-any.whl

dp/agent/cli/templates/ui/server/middleware.py

@@ -0,0 +1,43 @@
+"""
+FastAPI middleware definitions
+"""
+from fastapi import Request
+from fastapi.responses import PlainTextResponse
+from starlette.middleware.base import BaseHTTPMiddleware
+from server.utils import get_ak_info_from_request
+
+
+class RequestLoggingMiddleware(BaseHTTPMiddleware):
+    """Request middleware - for debugging"""
+    async def dispatch(self, request: Request, call_next):
+        try:
+            # Process request info and AK
+            access_key, _ = get_ak_info_from_request(request.headers)
+        except:
+            # Ignore any processing errors
+            pass
+        
+        response = await call_next(request)
+        return response
+
+
+class HostValidationMiddleware(BaseHTTPMiddleware):
+    """Host validation middleware"""
+    def __init__(self, app, allowed_hosts):
+        super().__init__(app)
+        self.allowed_hosts = allowed_hosts
+    
+    async def dispatch(self, request: Request, call_next):
+        host = request.headers.get("host", "").split(":")[0]
+        # If allowed list contains "*", allow all hosts
+        if "*" in self.allowed_hosts:
+            response = await call_next(request)
+            return response
+        # Otherwise check if host is in allowed list
+        if host and host not in self.allowed_hosts:
+            return PlainTextResponse(
+                content=f"Host '{host}' is not allowed",
+                status_code=403
+            )
+        response = await call_next(request)
+        return response

dp/agent/cli/templates/ui/server/models.py

@@ -0,0 +1,53 @@
+"""
+Data model definitions
+"""
+from dataclasses import dataclass, field
+from datetime import datetime
+from typing import Optional, List
+import uuid
+
+
+@dataclass
+class Message:
+    """Message data model"""
+    id: str
+    role: str  # 'user' or 'assistant' or 'tool'
+    content: str
+    timestamp: datetime = field(default_factory=datetime.now)
+    tool_name: Optional[str] = None
+    tool_status: Optional[str] = None
+
+
+@dataclass 
+class Session:
+    """Session data model"""
+    id: str
+    title: str = "New Session"
+    messages: List[Message] = field(default_factory=list)
+    created_at: datetime = field(default_factory=datetime.now)
+    last_message_at: datetime = field(default_factory=datetime.now)
+    is_modified: bool = False  # Mark if session has been modified
+    last_saved_at: Optional[datetime] = None  # Last save time
+    
+    def add_message(self, role: str, content: str, tool_name: Optional[str] = None, tool_status: Optional[str] = None):
+        """Add message to session"""
+        message = Message(
+            id=str(uuid.uuid4()),
+            role=role,
+            content=content,
+            tool_name=tool_name,
+            tool_status=tool_status
+        )
+        self.messages.append(message)
+        self.last_message_at = datetime.now()
+        self.is_modified = True  # Mark as modified
+        
+        if self.title == "New Session" and role == "user" and len(self.messages) <= 2:
+            self.title = content[:30] + "..." if len(content) > 30 else content
+        
+        return message
+    
+    def mark_saved(self):
+        """Mark session as saved"""
+        self.is_modified = False
+        self.last_saved_at = datetime.now()