ai_client 0.4.3 → 0.4.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 602b3197f09d1a0cb0525073ee456c54e6de6f835488d36c0ebf56259b6a72e9
-  data.tar.gz: 59c7d27f2c56fff21041cdd68ed09e3f3e624b7dead6059871dfe54cfcd46e58
+  metadata.gz: fc222406f840959b9987644d5f274b7422d7b157dc5ebaff6d22866a762d30f0
+  data.tar.gz: d534d1b7a975f1efe1d149fe558c8748c192bf41b4ee1a198d73488bb03a64e6
 SHA512:
-  metadata.gz: f98d17165dc760a2a4f81293a344b4edf65521e22228e8d908c809bd0c941c7c102c4b9de71779f7b9e2587278ded7e440b9b018fd4ef7054c4ab6d0ce2c563d
-  data.tar.gz: be3e0798c3894579417907732a9fd867d6d1eeae3de9dbe2dad1e89b2eb29146978fe015784920186773373b57dfff7fa8434a853e4e60668336066e2a9e1e81
+  metadata.gz: 5b988d95990e8b420aac8f4a30d5cdfa1ec80125d52d0ed0a48bae01d3c2e8a89a0e8307c7698ebb0818be4601f996618457fda25deeb07c608232e055e23bb8
+  data.tar.gz: ad9af7a235571f780873d1a0840fc89820e8e00a56b72e7ce9e0aa67626dcdcdaef306e0cb6333eb717320f53401d20e6ec72bdc876ba4d84492b3000f866fca

data/CHANGELOG.md

@@ -1,5 +1,11 @@
 ## [Unreleased]
 
+### [0.4.4] - 2025-03-02
+- Added ability to obtain a list of available models from an Ollama server.
+- Added `ollama_model_exists?` method to check if a specific model is available on an Ollama server.
+- Added command-line utility `bin/list_ollama_models` to display available Ollama models.
+- Added support for configuring custom Ollama hosts via the providers configuration.
+
 ### [0.4.3] - 2025-01-05
 - updated models.yml file with latest dump from open_router
 

data/Rakefile

@@ -9,6 +9,7 @@ rescue LoadError => error
   puts error.message
 end
 
+# Register the rake tasks
 Tocer::Rake::Register.call
 
 Minitest::TestTask.create

data/lib/ai_client/configuration.rb

@@ -156,13 +156,22 @@ class AiClient
         timeout: nil,
         return_raw: false,
         context_length: 5, # number of responses to add as context
-        providers: {},
+        providers: {
+          ollama: {
+            host: 'http://localhost:11434'
+          },
+          localai: {
+            host: 'http://localhost:8080'
+          }
+        },
         envar_api_key_names: {
           anthropic: ['ANTHROPIC_API_KEY'],
           google: ['GOOGLE_API_KEY'],
           mistral: ['MISTRAL_API_KEY'],
+          ollama: ['OLLAMA_API_KEY'],
           open_router: ['OPEN_ROUTER_API_KEY', 'OPENROUTER_API_KEY'],
-          openai: ['OPENAI_API_KEY']
+          openai: ['OPENAI_API_KEY'],
+          xai: ['XAI_API_KEY'],
         },
         provider_patterns: {
           anthropic: /^claude/i,
@@ -177,10 +186,10 @@ class AiClient
         default_model: {
           anthropic: 'claude-3-5-sonnet-20240620',
           openai: 'gpt-4o',
-          google: 'gemini-pro-1.5',
+          google: 'gemini-2.0-flash',
           mistral: 'mistral-large',
-          localai: 'llama3.2',
-          ollama: 'llama3.2',
+          localai: 'llama3.3',
+          ollama: 'llama3.3',
           open_router: 'auto'
         }
       )