101 lines
2.7 KiB
Rust
101 lines
2.7 KiB
Rust
// src/chat.rs
|
|
|
|
use seahorse::Context;
|
|
use std::process::Command;
|
|
use crate::config::ConfigPaths;
|
|
|
|
#[derive(Debug, Clone, PartialEq)]
|
|
pub enum Provider {
|
|
OpenAI,
|
|
Ollama,
|
|
}
|
|
|
|
impl Provider {
|
|
pub fn from_str(s: &str) -> Option<Self> {
|
|
match s.to_lowercase().as_str() {
|
|
"openai" => Some(Provider::OpenAI),
|
|
"ollama" => Some(Provider::Ollama),
|
|
_ => None,
|
|
}
|
|
}
|
|
|
|
pub fn as_str(&self) -> &'static str {
|
|
match self {
|
|
Provider::OpenAI => "openai",
|
|
Provider::Ollama => "ollama",
|
|
}
|
|
}
|
|
}
|
|
|
|
use std::fs;
|
|
use serde::Deserialize;
|
|
|
|
#[derive(Deserialize)]
|
|
struct OpenAIKey {
|
|
token: String,
|
|
}
|
|
|
|
fn load_openai_api_key() -> Option<String> {
|
|
let config = ConfigPaths::new();
|
|
let path = config.base_dir.join("openai.json");
|
|
let data = fs::read_to_string(path).ok()?;
|
|
let parsed: OpenAIKey = serde_json::from_str(&data).ok()?;
|
|
Some(parsed.token)
|
|
}
|
|
|
|
pub fn ask_chat(c: &Context, question: &str) {
|
|
let config = ConfigPaths::new();
|
|
let base_dir = config.base_dir.join("mcp");
|
|
let script_path = base_dir.join("scripts/ask.py");
|
|
|
|
let python_path = if cfg!(target_os = "windows") {
|
|
base_dir.join(".venv/Scripts/python.exe")
|
|
} else {
|
|
base_dir.join(".venv/bin/python")
|
|
};
|
|
|
|
let ollama_host = c.string_flag("host").ok();
|
|
let ollama_model = c.string_flag("model").ok();
|
|
let api_key = c.string_flag("api-key").ok()
|
|
.or_else(|| load_openai_api_key());
|
|
|
|
use crate::chat::Provider;
|
|
|
|
let provider_str = c.string_flag("provider").unwrap_or_else(|_| "ollama".to_string());
|
|
let provider = Provider::from_str(&provider_str).unwrap_or(Provider::Ollama);
|
|
|
|
println!("🔍 使用プロバイダー: {}", provider.as_str());
|
|
|
|
// 🛠️ command の定義をここで行う
|
|
let mut command = Command::new(python_path);
|
|
command.arg(script_path).arg(question);
|
|
|
|
// ✨ 環境変数をセット
|
|
command.env("PROVIDER", provider.as_str());
|
|
|
|
if let Some(host) = ollama_host {
|
|
command.env("OLLAMA_HOST", host);
|
|
}
|
|
if let Some(model) = ollama_model {
|
|
command.env("OLLAMA_MODEL", model);
|
|
}
|
|
if let Some(api_key) = api_key {
|
|
command.env("OPENAI_API_KEY", api_key);
|
|
}
|
|
|
|
// 🔁 実行
|
|
let output = command
|
|
.output()
|
|
.expect("❌ MCPチャットスクリプトの実行に失敗しました");
|
|
|
|
if output.status.success() {
|
|
println!("💬 {}", String::from_utf8_lossy(&output.stdout));
|
|
} else {
|
|
eprintln!(
|
|
"❌ 実行エラー: {}\n{}",
|
|
String::from_utf8_lossy(&output.stderr),
|
|
String::from_utf8_lossy(&output.stdout),
|
|
);
|
|
}
|
|
}
|