64d9dde209
- Added API endpoints for prompt optimization:
* POST /prompt/sessions: Create a new prompt optimization session
* GET /prompt/sessions/{session_id}: Retrieve session message history
* POST /prompt/sessions/{session_id}/messages: Send message and get optimized prompt
* PUT /prompt/model: Create or update system prompt model configuration
- Added database models for prompt optimization:
* prompt_opt_session: Stores session metadata
* prompt_opt_session_history: Stores session message history
* prompt_opt_message: Stores user and assistant messages
* prompt_opt_model_config: Stores system prompt model configurations
- Updated service layer to handle message creation, prompt optimization, and variable parsing
- Added corresponding Pydantic schemas for request and response validation
63 lines
1.8 KiB
Python
63 lines
1.8 KiB
Python
from .tenant_model import Tenants
|
|
from .user_model import User
|
|
from .workspace_model import Workspace, WorkspaceMember, WorkspaceRole
|
|
from .knowledge_model import Knowledge
|
|
from .document_model import Document
|
|
from .file_model import File
|
|
from .generic_file_model import GenericFile
|
|
from .models_model import ModelConfig, ModelProvider, ModelType, ModelApiKey
|
|
from .knowledgeshare_model import KnowledgeShare
|
|
from .app_model import App
|
|
from .agent_app_config_model import AgentConfig
|
|
from .app_release_model import AppRelease
|
|
from .memory_increment_model import MemoryIncrement
|
|
from .end_user_model import EndUser
|
|
from .appshare_model import AppShare
|
|
from .release_share_model import ReleaseShare
|
|
from .conversation_model import Conversation, Message
|
|
from .api_key_model import ApiKey, ApiKeyLog, ApiKeyType
|
|
from .data_config_model import DataConfig
|
|
from .multi_agent_model import MultiAgentConfig, AgentInvocation
|
|
from .workflow_model import WorkflowConfig, WorkflowExecution, WorkflowNodeExecution
|
|
from .retrieval_info import RetrievalInfo
|
|
from .prompt_optimizer_model import PromptOptimizerModelConfig, PromptOptimizerSession, PromptOptimizerSessionHistory
|
|
|
|
__all__ = [
|
|
"Tenants",
|
|
"User",
|
|
"Workspace",
|
|
"WorkspaceMember",
|
|
"WorkspaceRole",
|
|
"Knowledge",
|
|
"Document",
|
|
"File",
|
|
"GenericFile",
|
|
"ModelConfig",
|
|
"ModelProvider",
|
|
"ModelType",
|
|
"ModelApiKey",
|
|
"KnowledgeShare",
|
|
"App",
|
|
"AgentConfig",
|
|
"AppRelease",
|
|
"MemoryIncrement",
|
|
"EndUser",
|
|
"AppShare",
|
|
"ReleaseShare",
|
|
"Conversation",
|
|
"Message",
|
|
"ApiKey",
|
|
"ApiKeyLog",
|
|
"ApiKeyType",
|
|
"DataConfig",
|
|
"MultiAgentConfig",
|
|
"AgentInvocation",
|
|
"WorkflowConfig",
|
|
"WorkflowExecution",
|
|
"WorkflowNodeExecution",
|
|
"RetrievalInfo",
|
|
"PromptOptimizerModelConfig",
|
|
"PromptOptimizerSession",
|
|
"PromptOptimizerSessionHistory"
|
|
]
|