9e8db9dc81
This commit implements two major features: 1. Integration Search as Primary AI Data Source - Confluence, ServiceNow, and Azure DevOps searches execute before AI queries - Search results injected as system context for AI providers - Parallel search execution for performance - Webview-based fetch for HttpOnly cookie support - Persistent browser windows maintain authenticated sessions 2. AI Tool-Calling (Function Calling) - Allows AI to automatically execute functions during conversation - Implemented for OpenAI-compatible providers and Custom REST provider - Created add_ado_comment tool for updating Azure DevOps tickets - Iterative tool-calling loop supports multi-step workflows - Extensible architecture for adding new tools Key Files: - src-tauri/src/ai/tools.rs (NEW) - Tool definitions - src-tauri/src/integrations/*_search.rs (NEW) - Integration search modules - src-tauri/src/integrations/webview_fetch.rs (NEW) - HttpOnly cookie workaround - src-tauri/src/commands/ai.rs - Tool execution and integration search - src-tauri/src/ai/openai.rs - Tool-calling for OpenAI and Custom REST provider - All providers updated with tools parameter support Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>
70 lines
2.6 KiB
TypeScript
70 lines
2.6 KiB
TypeScript
import { create } from "zustand";
|
|
import { persist } from "zustand/middleware";
|
|
import type { AppSettings, ProviderConfig } from "@/lib/tauriCommands";
|
|
|
|
interface SettingsState extends AppSettings {
|
|
addProvider: (provider: ProviderConfig) => void;
|
|
updateProvider: (index: number, provider: ProviderConfig) => void;
|
|
removeProvider: (index: number) => void;
|
|
setProviders: (providers: ProviderConfig[]) => void;
|
|
setActiveProvider: (name: string) => void;
|
|
setTheme: (theme: "light" | "dark") => void;
|
|
getActiveProvider: () => ProviderConfig | undefined;
|
|
pii_enabled_patterns: Record<string, boolean>;
|
|
setPiiPattern: (id: string, enabled: boolean) => void;
|
|
}
|
|
|
|
export const useSettingsStore = create<SettingsState>()(
|
|
persist(
|
|
(set, get) => ({
|
|
theme: "dark" as const,
|
|
ai_providers: [] as ProviderConfig[],
|
|
active_provider: undefined,
|
|
default_provider: "ollama",
|
|
default_model: "llama3.2:3b",
|
|
ollama_url: "http://localhost:11434",
|
|
|
|
addProvider: (provider) =>
|
|
set((state) => ({ ai_providers: [...state.ai_providers, provider] })),
|
|
updateProvider: (index, provider) =>
|
|
set((state) => {
|
|
const providers = [...state.ai_providers];
|
|
providers[index] = provider;
|
|
return { ai_providers: providers };
|
|
}),
|
|
removeProvider: (index) =>
|
|
set((state) => ({
|
|
ai_providers: state.ai_providers.filter((_, i) => i !== index),
|
|
})),
|
|
setProviders: (providers) => set({ ai_providers: providers }),
|
|
setActiveProvider: (name) => set({ active_provider: name }),
|
|
setTheme: (theme) => set({ theme }),
|
|
pii_enabled_patterns: Object.fromEntries(
|
|
["email", "ip_address", "phone", "ssn", "credit_card", "hostname", "password", "api_key"]
|
|
.map((id) => [id, true])
|
|
) as Record<string, boolean>,
|
|
setPiiPattern: (id: string, enabled: boolean) =>
|
|
set((state) => ({
|
|
pii_enabled_patterns: { ...state.pii_enabled_patterns, [id]: enabled },
|
|
})),
|
|
getActiveProvider: () => {
|
|
const state = get();
|
|
return state.ai_providers.find((p) => p.name === state.active_provider)
|
|
?? state.ai_providers[0];
|
|
},
|
|
}),
|
|
{
|
|
name: "tftsr-settings",
|
|
// Don't persist ai_providers to localStorage - they're stored in encrypted database
|
|
partialize: (state) => ({
|
|
theme: state.theme,
|
|
active_provider: state.active_provider,
|
|
default_provider: state.default_provider,
|
|
default_model: state.default_model,
|
|
ollama_url: state.ollama_url,
|
|
pii_enabled_patterns: state.pii_enabled_patterns,
|
|
}),
|
|
}
|
|
)
|
|
);
|