diff --git a/apps/shinkai-desktop/src-tauri/src/local_shinkai_node/process_handlers/ollama_process_handler.rs b/apps/shinkai-desktop/src-tauri/src/local_shinkai_node/process_handlers/ollama_process_handler.rs index f5afcfe3a..c625ad9d4 100644 --- a/apps/shinkai-desktop/src-tauri/src/local_shinkai_node/process_handlers/ollama_process_handler.rs +++ b/apps/shinkai-desktop/src-tauri/src/local_shinkai_node/process_handlers/ollama_process_handler.rs @@ -13,7 +13,7 @@ pub struct OllamaOptions { pub ollama_host: String, pub ollama_num_parallel: String, pub ollama_max_loaded_models: String, - pub ollama_origins: Vec, + pub ollama_origins: String, } pub struct OllamaProcessHandler { @@ -41,7 +41,7 @@ impl OllamaProcessHandler { ollama_host: "127.0.0.1:11435".to_string(), ollama_num_parallel: "2".to_string(), ollama_max_loaded_models: "2".to_string(), - ollama_origins: ["*".to_string()].to_vec(), + ollama_origins: "*".to_string(), } } diff --git a/apps/shinkai-desktop/src/components/shinkai-node-manager/components/model-quality-tag.tsx b/apps/shinkai-desktop/src/components/shinkai-node-manager/components/model-quality-tag.tsx index 80831a12c..9377801ae 100644 --- a/apps/shinkai-desktop/src/components/shinkai-node-manager/components/model-quality-tag.tsx +++ b/apps/shinkai-desktop/src/components/shinkai-node-manager/components/model-quality-tag.tsx @@ -1,7 +1,7 @@ import { Badge } from '@shinkai_network/shinkai-ui'; import { cn } from '@shinkai_network/shinkai-ui/utils'; -import { OllamaModelQuality } from '../../../lib/shinkai-node-manager/ollama_models'; +import { OllamaModelQuality } from '../../../lib/shinkai-node-manager/ollama-models'; export const ModelQuailityTag = ({ quality, diff --git a/apps/shinkai-desktop/src/components/shinkai-node-manager/components/model-speed-tag.tsx b/apps/shinkai-desktop/src/components/shinkai-node-manager/components/model-speed-tag.tsx index f4496db09..bce83b0c4 100644 --- a/apps/shinkai-desktop/src/components/shinkai-node-manager/components/model-speed-tag.tsx +++ b/apps/shinkai-desktop/src/components/shinkai-node-manager/components/model-speed-tag.tsx @@ -1,7 +1,7 @@ import { Badge } from '@shinkai_network/shinkai-ui'; import { cn } from '@shinkai_network/shinkai-ui/utils'; -import { OllamaModelSpeed } from '../../../lib/shinkai-node-manager/ollama_models'; +import { OllamaModelSpeed } from '../../../lib/shinkai-node-manager/ollama-models'; export const ModelSpeedTag = ({ speed }: { speed: OllamaModelSpeed }) => { const emojiMap: { [key in OllamaModelSpeed]: string } = { diff --git a/apps/shinkai-desktop/src/components/shinkai-node-manager/ollama-models.tsx b/apps/shinkai-desktop/src/components/shinkai-node-manager/ollama-models.tsx index 943e62a89..5940ff3f3 100644 --- a/apps/shinkai-desktop/src/components/shinkai-node-manager/ollama-models.tsx +++ b/apps/shinkai-desktop/src/components/shinkai-node-manager/ollama-models.tsx @@ -17,12 +17,12 @@ import { ModelResponse, ProgressResponse } from 'ollama/browser'; import { useEffect, useState } from 'react'; import { toast } from 'sonner'; -import { OLLAMA_MODELS } from '../../lib/shinkai-node-manager/ollama_models'; import { useOllamaListQuery, useOllamaPullMutation, useOllamaRemoveMutation, } from '../../lib/shinkai-node-manager/ollama-client'; +import { OLLAMA_MODELS } from '../../lib/shinkai-node-manager/ollama-models'; import { useShinkaiNodeGetOllamaApiUrlQuery, useShinkaiNodeIsRunningQuery, @@ -35,7 +35,7 @@ import { ModelSpeedTag } from './components/model-speed-tag'; export const OllamaModels = () => { const auth = useAuth((auth) => auth.auth); const { data: ollamaApiUrl } = useShinkaiNodeGetOllamaApiUrlQuery(); - const ollamaConfig = { host: ollamaApiUrl || '' }; + const ollamaConfig = { host: ollamaApiUrl || 'http://127.0.0.1:11435' }; const { data: isShinkaiNodeRunning } = useShinkaiNodeIsRunningQuery(); const { mutateAsync: shinkaiNodeSpawn } = useShinkaiNodeSpawnMutation({}); const { mutateAsync: syncOllamaModels } = useSyncOllamaModels(); diff --git a/apps/shinkai-desktop/src/lib/shinkai-node-manager/ollama_models.ts b/apps/shinkai-desktop/src/lib/shinkai-node-manager/ollama-models.ts similarity index 100% rename from apps/shinkai-desktop/src/lib/shinkai-node-manager/ollama_models.ts rename to apps/shinkai-desktop/src/lib/shinkai-node-manager/ollama-models.ts