Spaces:
Running
Running
| export type GraphDb = "networkx" | "neo4j"; | |
| export type VectorDb = "postgresql" | "milvus"; | |
| export type DataDb = "postgresql" | "mongo"; | |
| export type ModelsMode = "local" | "remote"; | |
| export type ModelProvider = | |
| | "ollama" | |
| | "azure" | |
| | "openai" | |
| | "anthropic" | |
| | "deepseek" | |
| | "gcp_vertex" | |
| | "amazon_bedrock"; | |
| export type OcrMode = "docparser" | "docling"; | |
| export type PipelineMode = "accurate" | "lightweight"; | |
| export type SearchRetrieval = "hybrid" | "dense" | "bm25"; | |
| export type ServicesRuntime = "docker" | "manual"; | |
| export type PluginSource = "local" | "registry"; | |
| export function isPipelineMode(value: string): value is PipelineMode { | |
| return value === "accurate" || value === "lightweight"; | |
| } | |
| export interface DbChoices { | |
| vectorDb: VectorDb; | |
| dataDb: DataDb; | |
| graphDb: GraphDb; | |
| } | |
| export interface OllamaChoices { | |
| host: string; | |
| port: number; | |
| smallModel: string; | |
| largeModel: string; | |
| embeddingsLocalModel: string; | |
| } | |
| export interface GcpChoices { | |
| credentialsPath: string; | |
| projectId: string; | |
| smallLlmModel: string; | |
| largeLlmModel: string; | |
| embeddingModel: string; | |
| } | |
| export interface AzureChoices { | |
| smallLlmModel: string; | |
| largeLlmModel: string; | |
| llmApiVersion: string; | |
| llmEndpoint: string; | |
| llmSubscriptionKey: string; | |
| embeddingEndpoint: string; | |
| embeddingKey: string; | |
| embeddingModel: string; | |
| } | |
| export interface OpenAIChoices { | |
| apiKey: string; | |
| baseUrl?: string; | |
| smallLlmModel: string; | |
| largeLlmModel: string; | |
| embeddingModel: string; | |
| } | |
| export interface BedrockChoices { | |
| region: string; | |
| accessKeyId: string; | |
| secretAccessKey: string; | |
| sessionToken?: string; | |
| smallLlmModel: string; | |
| largeLlmModel: string; | |
| embeddingModel: string; | |
| } | |
| export interface AnthropicChoices { | |
| apiKey: string; | |
| smallLlmModel: string; | |
| largeLlmModel: string; | |
| } | |
| export interface DeepSeekChoices { | |
| apiKey: string; | |
| smallLlmModel: string; | |
| largeLlmModel: string; | |
| } | |
| export interface EmbeddingDimensions { | |
| nodesDimension: number; | |
| tripletsDimension: number; | |
| observationsDimension: number; | |
| dataDimension: number; | |
| relationshipsDimension: number; | |
| } | |
| export interface ModelsChoices { | |
| mode: ModelsMode; | |
| llmSmallProvider: ModelProvider; | |
| llmLargeProvider: ModelProvider; | |
| embeddingsProvider: ModelProvider; | |
| embeddingDimensions?: EmbeddingDimensions; | |
| ollama?: OllamaChoices; | |
| gcp?: GcpChoices; | |
| azure?: AzureChoices; | |
| openai?: OpenAIChoices; | |
| anthropic?: AnthropicChoices; | |
| deepseek?: DeepSeekChoices; | |
| bedrock?: BedrockChoices; | |
| } | |
| export interface PostgresConnection { | |
| host: string; | |
| port: number; | |
| username: string; | |
| password: string; | |
| systemDatabase: string; | |
| maintenanceDatabase: string; | |
| } | |
| export interface Neo4jConnection { | |
| host: string; | |
| port: number; | |
| username: string; | |
| password: string; | |
| } | |
| export interface MilvusConnection { | |
| host: string; | |
| port: number; | |
| uri?: string; | |
| token?: string; | |
| } | |
| export interface MongoConnection { | |
| host: string; | |
| port: number; | |
| username: string; | |
| password: string; | |
| } | |
| export interface RedisConnection { | |
| host: string; | |
| port: number; | |
| } | |
| export interface Connections { | |
| redis: RedisConnection; | |
| postgresql?: PostgresConnection; | |
| neo4j?: Neo4jConnection; | |
| milvus?: MilvusConnection; | |
| mongo?: MongoConnection; | |
| } | |
| export interface AuthChoices { | |
| brainpatToken: string; | |
| } | |
| export interface PipelineChoices { | |
| ocrMode: OcrMode; | |
| } | |
| export interface SearchChoices { | |
| enabled: boolean; | |
| retrieval: SearchRetrieval; | |
| } | |
| export interface PluginChoice { | |
| name: string; | |
| source: PluginSource; | |
| version?: string; | |
| path?: string; | |
| } | |
| export interface InitChoices { | |
| dbs: DbChoices; | |
| models: ModelsChoices; | |
| pipeline: PipelineChoices; | |
| search: SearchChoices; | |
| connections: Connections; | |
| auth: AuthChoices; | |
| plugins: PluginChoice[]; | |
| servicesRuntime: ServicesRuntime; | |
| usedDefaults: boolean; | |
| } | |
| export interface InstallState { | |
| cloned: boolean; | |
| venvCreated: boolean; | |
| depsInstalled: boolean; | |
| envWritten: boolean; | |
| containersStarted: boolean; | |
| selectedServices: string[] | null; | |
| installedPlugins: string[] | null; | |
| servicesRuntime: ServicesRuntime | null; | |
| sourcePath: string; | |
| repoUrl: string; | |
| branch: string; | |
| } | |
| export type Platform = "darwin" | "linux" | "win32"; | |