#include <ollama_ai_service.h>
Public Member Functions | |
OllamaAIService (const OllamaConfig &config) | |
void | SetRomContext (Rom *rom) override |
absl::StatusOr< AgentResponse > | GenerateResponse (const std::string &prompt) override |
absl::StatusOr< AgentResponse > | GenerateResponse (const std::vector< agent::ChatMessage > &history) override |
absl::Status | CheckAvailability () |
absl::StatusOr< std::vector< std::string > > | ListAvailableModels () |
![]() | |
virtual | ~AIService ()=default |
Private Member Functions | |
std::string | BuildSystemPrompt () |
absl::StatusOr< std::string > | ParseOllamaResponse (const std::string &json_response) |
Private Attributes | |
OllamaConfig | config_ |
PromptBuilder | prompt_builder_ |
Definition at line 25 of file ollama_ai_service.h.
|
explicit |
Definition at line 18 of file ollama_ai_service.cc.
References yaze::cli::PromptBuilder::BuildSystemInstructionWithExamples(), BuildSystemPrompt(), config_, yaze::cli::PromptBuilder::LoadResourceCatalogue(), prompt_builder_, yaze::cli::OllamaConfig::system_prompt, and yaze::cli::OllamaConfig::use_enhanced_prompting.
|
overridevirtual |
Reimplemented from yaze::cli::AIService.
Definition at line 41 of file ollama_ai_service.cc.
References prompt_builder_, and yaze::cli::PromptBuilder::SetRom().
|
overridevirtual |
Implements yaze::cli::AIService.
Definition at line 159 of file ollama_ai_service.cc.
References GenerateResponse(), and yaze::cli::agent::ChatMessage::kUser.
Referenced by GenerateResponse().
|
overridevirtual |
Implements yaze::cli::AIService.
Definition at line 164 of file ollama_ai_service.cc.
References yaze::cli::ToolCall::args, yaze::cli::OllamaConfig::base_url, yaze::cli::PromptBuilder::BuildPromptFromHistory(), yaze::cli::AgentResponse::commands, config_, yaze::cli::OllamaConfig::max_tokens, yaze::cli::OllamaConfig::model, prompt_builder_, yaze::cli::AgentResponse::reasoning, yaze::cli::OllamaConfig::system_prompt, yaze::cli::OllamaConfig::temperature, yaze::cli::AgentResponse::text_response, yaze::cli::AgentResponse::tool_calls, and yaze::cli::ToolCall::tool_name.
absl::Status yaze::cli::OllamaAIService::CheckAvailability | ( | ) |
Definition at line 45 of file ollama_ai_service.cc.
References yaze::cli::OllamaConfig::base_url, config_, and yaze::cli::OllamaConfig::model.
absl::StatusOr< std::vector< std::string > > yaze::cli::OllamaAIService::ListAvailableModels | ( | ) |
Definition at line 104 of file ollama_ai_service.cc.
References yaze::cli::OllamaConfig::base_url, and config_.
|
private |
Definition at line 35 of file ollama_ai_service.cc.
References yaze::cli::PromptBuilder::BuildSystemInstruction(), and prompt_builder_.
Referenced by OllamaAIService().
|
private |
Definition at line 138 of file ollama_ai_service.cc.
|
private |
Definition at line 44 of file ollama_ai_service.h.
Referenced by CheckAvailability(), GenerateResponse(), ListAvailableModels(), and OllamaAIService().
|
private |
Definition at line 45 of file ollama_ai_service.h.
Referenced by BuildSystemPrompt(), GenerateResponse(), OllamaAIService(), and SetRomContext().