mirror of
https://github.com/GNS3/gns3-server.git
synced 2026-09-02 08:05:14 +03:00
4a017b1a7f
- Add `pinned` column to chat_sessions table with default FALSE
- Implement database migration for existing installations using PRAGMA table_info
- Create composite index for pinned + updated_at sorting
- Add pin/unpin API endpoints (PUT/DELETE /sessions/{id}/pin)
- Update session listing to sort by pinned status then updated_at
- Extend ChatSessionsRepository with pin_session method
- Update API documentation to reflect new pinning functionality
The feature allows users to pin important chat sessions to the top of the list. Sessions are sorted with pinned sessions first (by updated_at), followed by regular sessions (by updated_at). Database migration ensures backward compatibility with existing installations.
113 lines
5.4 KiB
Python
113 lines
5.4 KiB
Python
#
|
|
# Copyright (C) 2025 GNS3 Technologies Inc.
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
"""
|
|
Chat API schemas for GNS3 Copilot integration.
|
|
"""
|
|
|
|
from pydantic import BaseModel, Field
|
|
from typing import Optional, List, Dict, Any, Literal
|
|
|
|
|
|
class OpenAIToolCall(BaseModel):
|
|
"""Tool call information (OpenAI compatible format)."""
|
|
|
|
id: str = Field(..., description="Tool call ID")
|
|
type: Literal["function"] = Field(default="function", description="Tool call type")
|
|
function: Dict[str, Any] = Field(..., description="Function name and arguments")
|
|
|
|
|
|
class ChatRequest(BaseModel):
|
|
"""Chat request model."""
|
|
|
|
message: str = Field(..., description="User message content")
|
|
session_id: Optional[str] = Field(None, description="Session ID (auto-generated if not provided)")
|
|
stream: bool = Field(default=True, description="Enable streaming response")
|
|
temperature: Optional[float] = Field(None, description="LLM temperature parameter")
|
|
mode: Literal["text"] = Field(default="text", description="Interaction mode")
|
|
|
|
|
|
class ChatResponse(BaseModel):
|
|
"""Chat streaming response model."""
|
|
|
|
type: Literal[
|
|
"content", # AI text content
|
|
"tool_call", # Tool call request
|
|
"tool_start", # Tool execution started
|
|
"tool_end", # Tool execution completed
|
|
"error", # Error message
|
|
"done", # Stream ended
|
|
"heartbeat" # Keep-alive signal
|
|
] = Field(..., description="Response message type")
|
|
content: Optional[str] = Field(None, description="Text content (for type=content)")
|
|
message_id: Optional[str] = Field(None, description="Message ID")
|
|
tool_call: Optional[OpenAIToolCall] = Field(None, description="Tool call (for type=tool_call)")
|
|
tool_name: Optional[str] = Field(None, description="Tool name (for type=tool_start/end)")
|
|
tool_output: Optional[str] = Field(None, description="Tool output (for type=tool_end)")
|
|
error: Optional[str] = Field(None, description="Error message (for type=error)")
|
|
session_id: Optional[str] = Field(None, description="Session ID (for type=heartbeat/done)")
|
|
|
|
|
|
class OpenAIMessage(BaseModel):
|
|
"""Message model for conversation history."""
|
|
|
|
id: str = Field(..., description="Message ID")
|
|
role: Literal["user", "assistant", "system", "tool"] = Field(..., description="Message role")
|
|
content: str = Field(..., description="Message content")
|
|
name: Optional[str] = Field(None, description="Tool message name")
|
|
tool_call_id: Optional[str] = Field(None, description="Associated tool call ID (for tool messages)")
|
|
tool_calls: Optional[List[OpenAIToolCall]] = Field(None, description="Tool calls (for assistant messages)")
|
|
metadata: Dict[str, Any] = Field(default_factory=dict, description="Message metadata")
|
|
created_at: str = Field(..., description="Message timestamp (ISO 8601)")
|
|
|
|
|
|
class ConversationHistory(BaseModel):
|
|
"""Conversation history model."""
|
|
|
|
thread_id: str = Field(..., description="Thread/session ID")
|
|
title: str = Field(..., description="Conversation title")
|
|
messages: List[OpenAIMessage] = Field(default_factory=list, description="Conversation messages")
|
|
created_at: Optional[str] = Field(None, description="Creation timestamp (ISO 8601)")
|
|
updated_at: Optional[str] = Field(None, description="Last update timestamp (ISO 8601)")
|
|
llm_calls: int = Field(default=0, description="Total LLM calls in this conversation")
|
|
|
|
|
|
class ChatSession(BaseModel):
|
|
"""Chat session model."""
|
|
|
|
id: Optional[int] = Field(None, description="Database ID")
|
|
thread_id: str = Field(..., description="Thread/session ID")
|
|
user_id: str = Field(..., description="User ID")
|
|
project_id: str = Field(..., description="Associated GNS3 project ID")
|
|
title: str = Field(..., description="Session title")
|
|
message_count: int = Field(default=0, description="Number of messages")
|
|
llm_calls_count: int = Field(default=0, description="Number of LLM calls")
|
|
input_tokens: int = Field(default=0, description="Input tokens used")
|
|
output_tokens: int = Field(default=0, description="Output tokens generated")
|
|
total_tokens: int = Field(default=0, description="Total tokens used")
|
|
last_message_at: Optional[str] = Field(None, description="Last message timestamp (ISO 8601)")
|
|
created_at: Optional[str] = Field(None, description="Creation timestamp (ISO 8601)")
|
|
updated_at: Optional[str] = Field(None, description="Last update timestamp (ISO 8601)")
|
|
metadata: Dict[str, Any] = Field(default_factory=dict, description="Session metadata")
|
|
stats: Dict[str, Any] = Field(default_factory=dict, description="Session statistics")
|
|
pinned: bool = Field(default=False, description="Whether the session is pinned to the top")
|
|
|
|
|
|
class RenameSession(BaseModel):
|
|
"""Rename session request model."""
|
|
|
|
title: str = Field(..., description="New session title", min_length=1, max_length=255)
|