mirror of
https://github.com/crawlab-team/crawlab.git
synced 2026-01-21 17:21:09 +01:00
ec5bcf9287
- Added AutoProbe interfaces to define models for autoprobe functionality, including AutoProbe, AutoProbeTask, and AutoProbeFetchResult. - Created AutoProbeStore module for managing state, getters, mutations, and actions related to autoprobe. - Updated existing LLM interfaces to utilize LLMResponseUsage for token tracking. - Refactored store index to include autoprobe namespace and removed obsolete AiStore module.
139 lines
3.2 KiB
Go
139 lines
3.2 KiB
Go
export declare global {
|
|
interface LLMResponseUsage {
|
|
prompt_tokens?: number;
|
|
completion_tokens?: number;
|
|
total_tokens?: number;
|
|
}
|
|
|
|
type LLMProviderType =
|
|
| 'openai'
|
|
| 'azure-openai'
|
|
| 'anthropic'
|
|
| 'gemini'
|
|
| 'grok'
|
|
| 'qwen'
|
|
| 'deepseek'
|
|
| 'mistral'
|
|
| 'openai-compatible';
|
|
|
|
interface LLMProvider extends BaseModel {
|
|
type?: LLMProviderType;
|
|
name?: string;
|
|
enabled?: boolean;
|
|
api_key?: string;
|
|
api_base_url?: string;
|
|
api_version?: string;
|
|
models?: string[];
|
|
}
|
|
|
|
interface LLMProviderItem {
|
|
type: LLMProviderType;
|
|
name: string;
|
|
icon?: Icon;
|
|
defaultModels?: string[];
|
|
defaultApiVersions?: string[];
|
|
}
|
|
|
|
interface LLMProviderModel {
|
|
providerId?: string;
|
|
model?: string;
|
|
}
|
|
|
|
type ChatMessageRole = 'system' | 'user' | 'assistant';
|
|
|
|
type ChatMessageStatus = 'pending' | 'completed' | 'failed';
|
|
|
|
interface ChatMessage extends BaseModel {
|
|
conversation_id: string;
|
|
role: ChatMessageRole;
|
|
content?: string;
|
|
content_ids?: string[];
|
|
contents?: ChatMessageContent[];
|
|
tokens?: number;
|
|
model?: string;
|
|
metadata?: Record<string, any>;
|
|
status: ChatMessageStatus;
|
|
error?: string;
|
|
usage?: LLMResponseUsage;
|
|
|
|
// Frontend UI-specific properties
|
|
timestamp?: Date;
|
|
isStreaming?: boolean;
|
|
}
|
|
|
|
type ChatMessageContentType = 'text' | 'action';
|
|
type ChatMessageActionStatus = 'pending' | 'success' | 'failed';
|
|
|
|
interface ChatMessageContent extends BaseModel {
|
|
message_id?: string;
|
|
key?: string;
|
|
parameters?: Record<string, any>;
|
|
content?: string;
|
|
type: ChatMessageContentType;
|
|
action?: string;
|
|
action_target?: string;
|
|
action_status?: ChatMessageActionStatus;
|
|
hidden?: boolean;
|
|
usage?: LLMResponseUsage;
|
|
|
|
// Frontend UI-specific properties
|
|
isStreaming?: boolean;
|
|
}
|
|
|
|
type ChatConversationStatus = 'active' | 'archived' | 'deleted';
|
|
|
|
interface ChatConversation extends BaseModel {
|
|
title?: string;
|
|
description?: string;
|
|
user_id?: string;
|
|
model?: string;
|
|
status?: ChatConversationStatus;
|
|
last_message_at?: string;
|
|
settings?: Record<string, any>;
|
|
tags?: string[];
|
|
messages?: ChatMessage[];
|
|
created_at?: string;
|
|
updated_at?: string;
|
|
}
|
|
|
|
interface ChatRequest {
|
|
provider_id: string;
|
|
model: string;
|
|
query: string;
|
|
system_prompt?: string;
|
|
max_tokens?: number;
|
|
temperature?: number;
|
|
top_p?: number;
|
|
other_params?: Record<string, any>;
|
|
conversation_id?: string;
|
|
}
|
|
|
|
interface ChatbotConfig {
|
|
providerId?: string;
|
|
model?: string;
|
|
systemPrompt?: string;
|
|
temperature?: number;
|
|
maxTokens?: number;
|
|
}
|
|
|
|
interface ChatbotStreamMessage {
|
|
conversation_id?: string;
|
|
conversation_title?: string;
|
|
message_id?: string;
|
|
key?: string;
|
|
parameters?: Record<string, any>;
|
|
content?: string;
|
|
type: 'text' | 'action'; // Message type
|
|
action_id?: string;
|
|
action?: string;
|
|
action_target?: string;
|
|
action_status?: ChatMessageActionStatus;
|
|
is_done?: boolean;
|
|
is_initial?: boolean;
|
|
error?: string;
|
|
hidden?: boolean;
|
|
is_text_done?: boolean;
|
|
usage?: LLMResponseUsage;
|
|
}
|
|
}
|