mirror of
https://github.com/memohai/Memoh.git
synced 2026-04-27 07:16:19 +09:00
BBQ
1c15eb2146
- Replace chat package with conversation flow architecture - Add channel identity avatar support (migration 0002) - Refactor channel adapters, identities, and message routing - Update frontend: simplify composables, modernize UI components - Improve Docker builds with cache mounts and version metadata - Optimize healthchecks and simplify service dependencies
101 lines
2.2 KiB
Go
101 lines
2.2 KiB
Go
package models
|
|
|
|
import (
|
|
"errors"
|
|
|
|
"github.com/google/uuid"
|
|
)
|
|
|
|
type ModelType string
|
|
|
|
const (
|
|
ModelTypeChat ModelType = "chat"
|
|
ModelTypeEmbedding ModelType = "embedding"
|
|
)
|
|
|
|
const (
|
|
ModelInputText = "text"
|
|
ModelInputImage = "image"
|
|
)
|
|
|
|
type ClientType string
|
|
|
|
const (
|
|
ClientTypeOpenAI ClientType = "openai"
|
|
ClientTypeOpenAICompat ClientType = "openai-compat"
|
|
ClientTypeAnthropic ClientType = "anthropic"
|
|
ClientTypeGoogle ClientType = "google"
|
|
ClientTypeAzure ClientType = "azure"
|
|
ClientTypeBedrock ClientType = "bedrock"
|
|
ClientTypeMistral ClientType = "mistral"
|
|
ClientTypeXAI ClientType = "xai"
|
|
ClientTypeOllama ClientType = "ollama"
|
|
ClientTypeDashscope ClientType = "dashscope"
|
|
)
|
|
|
|
type Model struct {
|
|
ModelID string `json:"model_id"`
|
|
Name string `json:"name"`
|
|
LlmProviderID string `json:"llm_provider_id"`
|
|
IsMultimodal bool `json:"is_multimodal"`
|
|
Input []string `json:"input"`
|
|
Type ModelType `json:"type"`
|
|
Dimensions int `json:"dimensions"`
|
|
}
|
|
|
|
func (m *Model) Validate() error {
|
|
if m.ModelID == "" {
|
|
return errors.New("model ID is required")
|
|
}
|
|
if m.LlmProviderID == "" {
|
|
return errors.New("llm provider ID is required")
|
|
}
|
|
if _, err := uuid.Parse(m.LlmProviderID); err != nil {
|
|
return errors.New("llm provider ID must be a valid UUID")
|
|
}
|
|
if m.Type != ModelTypeChat && m.Type != ModelTypeEmbedding {
|
|
return errors.New("invalid model type")
|
|
}
|
|
if m.Type == ModelTypeEmbedding && m.Dimensions <= 0 {
|
|
return errors.New("dimensions must be greater than 0")
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
type AddRequest Model
|
|
|
|
type AddResponse struct {
|
|
ID string `json:"id"`
|
|
ModelID string `json:"model_id"`
|
|
}
|
|
|
|
type GetRequest struct {
|
|
ID string `json:"id"`
|
|
}
|
|
|
|
type GetResponse struct {
|
|
ModelId string `json:"model_id"`
|
|
Model
|
|
}
|
|
|
|
type UpdateRequest Model
|
|
|
|
type ListRequest struct {
|
|
Type ModelType `json:"type,omitempty"`
|
|
ClientType ClientType `json:"client_type,omitempty"`
|
|
}
|
|
|
|
type DeleteRequest struct {
|
|
ID string `json:"id,omitempty"`
|
|
ModelID string `json:"model_id,omitempty"`
|
|
}
|
|
|
|
type DeleteResponse struct {
|
|
Message string `json:"message"`
|
|
}
|
|
|
|
type CountResponse struct {
|
|
Count int64 `json:"count"`
|
|
}
|