llm_conductor 1.1.2 → 1.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 0ca46783dd713d49b3292342d83f5adde4a0da684e4365004651464e6ac630bb
-  data.tar.gz: 5f693e2e4d8da70bebe5faf174a71e1880a705eb7dc6468d4fdf6774b8f2e9f3
+  metadata.gz: c6ed179bb9142839bcc6feab8d06d61c27ff8279406bc7839f6d09ba14cb573f
+  data.tar.gz: a8ca32fecd9ac81326f7cefcf482f1b6a110b78ca2168c1c8ccbde5e034becb3
 SHA512:
-  metadata.gz: 70ccb3ae2317588199a2820f1da19188e5ada27da19052de9cc964aeac775bbdd8feae350c958e797676eb8cbaf6549966591ad1299a1088ceeb8cfcbf70dc35
-  data.tar.gz: d24c1ff9423b009b2d50a35d6b122817174d5871813db037ee15378938cfd88aaf9dbc4765325bd65cf99d9fbbbeaae2a2d5ad926677efc6b36187209ddccf09
+  metadata.gz: 581da83914c51a3966010d03491c3f57be4ed393bb572f2fdc9d0205f8680f4891f2b058ecf7642ea7bf26bea452a976946b6198d0419afb2e771de3bc112aea
+  data.tar.gz: 00eb70033cb739b7236b759a30219eb5eb6b72db7bba6c7ee519b98cf186e799cbf4f8696acf237d19a8fbfcca97dd9a189ce4f3b4f8f3d8a7d9ff1729d7eb86

data/.rubocop.yml

@@ -29,11 +29,15 @@ Style/HashSyntax:
 Lint/ConstantDefinitionInBlock:
   Enabled: false
 
+Metrics/ClassLength:
+  Max: 120
+
 Metrics/MethodLength:
   Max: 15
   Exclude:
     - 'lib/llm_conductor/prompts.rb'
     - 'lib/llm_conductor/clients/openrouter_client.rb'
+    - 'lib/llm_conductor/clients/zai_client.rb'
 
 RSpec/ExampleLength:
   Enabled: false
@@ -91,19 +95,23 @@ Metrics/AbcSize:
   Exclude:
     - 'lib/llm_conductor/prompts.rb'
     - 'lib/llm_conductor/clients/openrouter_client.rb'
+    - 'lib/llm_conductor/clients/zai_client.rb'
 
 Metrics/CyclomaticComplexity:
   Exclude:
+    - 'lib/llm_conductor.rb'
     - 'lib/llm_conductor/prompts.rb'
     - 'lib/llm_conductor/clients/openrouter_client.rb'
+    - 'lib/llm_conductor/clients/zai_client.rb'
 
 Metrics/PerceivedComplexity:
   Exclude:
     - 'lib/llm_conductor/prompts.rb'
     - 'lib/llm_conductor/clients/openrouter_client.rb'
+    - 'lib/llm_conductor/clients/zai_client.rb'
 
 Layout/LineLength:
-  Max: 120
+  Max: 125
 
 # Performance cops (from .rubocop_todo.yml)
 Performance/RedundantEqualityComparisonBlock:

data/README.md

@@ -1,12 +1,12 @@
 # LLM Conductor
 
-A powerful Ruby gem from [Ekohe](https://ekohe.com) for orchestrating multiple Language Model providers with a unified, modern interface. LLM Conductor provides seamless integration with OpenAI GPT, Anthropic Claude, Google Gemini, Groq, Ollama, and OpenRouter with advanced prompt management, data building patterns, vision/multimodal support, and comprehensive response handling.
+A powerful Ruby gem from [Ekohe](https://ekohe.com) for orchestrating multiple Language Model providers with a unified, modern interface. LLM Conductor provides seamless integration with OpenAI GPT, Anthropic Claude, Google Gemini, Groq, Ollama, OpenRouter, and Z.ai (Zhipu AI) with advanced prompt management, data building patterns, vision/multimodal support, and comprehensive response handling.
 
 ## Features
 
-🚀 **Multi-Provider Support** - OpenAI GPT, Anthropic Claude, Google Gemini, Groq, Ollama, and OpenRouter with automatic vendor detection
+🚀 **Multi-Provider Support** - OpenAI GPT, Anthropic Claude, Google Gemini, Groq, Ollama, OpenRouter, and Z.ai with automatic vendor detection
 🎯 **Unified Modern API** - Simple `LlmConductor.generate()` interface with rich Response objects  
-🖼️ **Vision/Multimodal Support** - Send images alongside text prompts for vision-enabled models (OpenRouter)
+🖼️ **Vision/Multimodal Support** - Send images alongside text prompts for vision-enabled models (OpenRouter, Z.ai GLM-4.5V)
 📝 **Advanced Prompt Management** - Registrable prompt classes with inheritance and templating  
 🏗️ **Data Builder Pattern** - Structured data preparation for complex LLM inputs  
 ⚡ **Smart Configuration** - Rails-style configuration with environment variable support  
@@ -120,6 +120,11 @@ LlmConductor.configure do |config|
     uri_base: 'https://openrouter.ai/api/v1' # Optional, this is the default
   )
 
+  config.zai(
+    api_key: ENV['ZAI_API_KEY'],
+    uri_base: 'https://api.z.ai/api/paas/v4' # Optional, this is the default
+  )
+
   # Optional: Configure custom logger
   config.logger = Logger.new($stdout)                  # Log to stdout
   config.logger = Logger.new('log/llm_conductor.log')  # Log to file
@@ -160,6 +165,7 @@ The gem automatically detects these environment variables:
 - `GROQ_API_KEY` - Groq API key
 - `OLLAMA_ADDRESS` - Ollama server address
 - `OPENROUTER_API_KEY` - OpenRouter API key
+- `ZAI_API_KEY` - Z.ai (Zhipu AI) API key
 
 ## Supported Providers & Models
 
@@ -309,6 +315,81 @@ LlmConductor.configure do |config|
 end
 ```
 
+### Z.ai (Zhipu AI) - GLM Models with Vision Support
+Z.ai provides access to GLM (General Language Model) series including the powerful GLM-4.5V multimodal model with 64K context window and vision capabilities.
+
+**Text models:**
+- `glm-4-plus` - Enhanced text-only model
+- `glm-4` - Standard GLM-4 model
+
+**Vision-capable models:**
+- `glm-4.5v` - Latest multimodal model with 64K context ✅ **RECOMMENDED**
+- `glm-4v` - Previous generation vision model
+
+```ruby
+# Text-only request with GLM-4-plus
+response = LlmConductor.generate(
+  model: 'glm-4-plus',
+  vendor: :zai,
+  prompt: 'Explain quantum computing in simple terms'
+)
+
+# Vision request with GLM-4.5V - single image
+response = LlmConductor.generate(
+  model: 'glm-4.5v',
+  vendor: :zai,
+  prompt: {
+    text: 'What is in this image?',
+    images: 'https://example.com/image.jpg'
+  }
+)
+
+# Vision request with multiple images
+response = LlmConductor.generate(
+  model: 'glm-4.5v',
+  vendor: :zai,
+  prompt: {
+    text: 'Compare these images and identify differences',
+    images: [
+      'https://example.com/image1.jpg',
+      'https://example.com/image2.jpg'
+    ]
+  }
+)
+
+# Vision request with detail level
+response = LlmConductor.generate(
+  model: 'glm-4.5v',
+  vendor: :zai,
+  prompt: {
+    text: 'Analyze this document in detail',
+    images: [
+      { url: 'https://example.com/document.jpg', detail: 'high' }
+    ]
+  }
+)
+
+# Base64 encoded local images
+require 'base64'
+image_data = Base64.strict_encode64(File.read('path/to/image.jpg'))
+response = LlmConductor.generate(
+  model: 'glm-4.5v',
+  vendor: :zai,
+  prompt: {
+    text: 'What is in this image?',
+    images: "data:image/jpeg;base64,#{image_data}"
+  }
+)
+```
+
+**GLM-4.5V Features:**
+- 64K token context window
+- Multimodal understanding (text + images)
+- Document understanding and OCR
+- Image reasoning and analysis
+- Base64 image support for local files
+- OpenAI-compatible API format
+
 ### Vendor Detection
 
 The gem automatically detects the appropriate provider based on model names:
@@ -316,6 +397,7 @@ The gem automatically detects the appropriate provider based on model names:
 - **OpenAI**: Models starting with `gpt-` (e.g., `gpt-4`, `gpt-3.5-turbo`)
 - **Anthropic**: Models starting with `claude-` (e.g., `claude-3-5-sonnet-20241022`)
 - **Google Gemini**: Models starting with `gemini-` (e.g., `gemini-2.5-flash`, `gemini-2.0-flash`)
+- **Z.ai**: Models starting with `glm-` (e.g., `glm-4.5v`, `glm-4-plus`, `glm-4v`)
 - **Groq**: Models starting with `llama`, `mixtral`, `gemma`, or `qwen` (e.g., `llama-3.1-70b-versatile`, `mixtral-8x7b-32768`, `gemma-7b-it`, `qwen-2.5-72b-instruct`)
 - **Ollama**: All other models (e.g., `llama3.2`, `mistral`, `codellama`)
 
@@ -569,6 +651,8 @@ Check the `/examples` directory for comprehensive usage examples:
 - `rag_usage.rb` - RAG implementation examples
 - `gemini_usage.rb` - Google Gemini integration
 - `groq_usage.rb` - Groq integration with various models
+- `openrouter_vision_usage.rb` - OpenRouter vision/multimodal examples
+- `zai_usage.rb` - Z.ai GLM-4.5V vision and text examples
 
 ## Development
 

data/VISION_USAGE.md

@@ -1,9 +1,11 @@
 # Vision/Multimodal Usage Guide
 
-This guide explains how to use vision/multimodal capabilities with the OpenRouter client in LLM Conductor.
+This guide explains how to use vision/multimodal capabilities with the OpenRouter and Z.ai clients in LLM Conductor.
 
 ## Quick Start
 
+### Using OpenRouter
+
 ```ruby
 require 'llm_conductor'
 
@@ -25,8 +27,33 @@ response = LlmConductor.generate(
 puts response.output
 ```
 
+### Using Z.ai (Zhipu AI)
+
+```ruby
+require 'llm_conductor'
+
+# Configure
+LlmConductor.configure do |config|
+  config.zai(api_key: ENV['ZAI_API_KEY'])
+end
+
+# Analyze an image with GLM-4.5V
+response = LlmConductor.generate(
+  model: 'glm-4.5v',
+  vendor: :zai,
+  prompt: {
+    text: 'What is in this image?',
+    images: 'https://example.com/image.jpg'
+  }
+)
+
+puts response.output
+```
+
 ## Recommended Models
 
+### OpenRouter Models
+
 For vision tasks via OpenRouter, these models work reliably:
 
 - **`openai/gpt-4o-mini`** - Fast, reliable, good balance of cost/quality ✅
@@ -34,6 +61,14 @@ For vision tasks via OpenRouter, these models work reliably:
 - **`anthropic/claude-3.5-sonnet`** - High quality analysis
 - **`openai/gpt-4o`** - Best quality (higher cost)
 
+### Z.ai Models (Zhipu AI)
+
+For vision tasks via Z.ai, these GLM models are recommended:
+
+- **`glm-4.5v`** - GLM-4.5V multimodal model (64K context window) ✅
+- **`glm-4-plus`** - Text-only model with enhanced capabilities
+- **`glm-4v`** - Previous generation vision model
+
 ## Usage Formats
 
 ### 1. Single Image (Simple Format)
@@ -169,11 +204,18 @@ response = LlmConductor.generate(
 
 ### Run Examples
 
+For OpenRouter:
 ```bash
 export OPENROUTER_API_KEY='your-key'
 ruby examples/openrouter_vision_usage.rb
 ```
 
+For Z.ai:
+```bash
+export ZAI_API_KEY='your-key'
+ruby examples/zai_usage.rb
+```
+
 ## Token Counting
 
 Token counting for multimodal requests counts only the text portion. Image tokens vary by:
@@ -223,11 +265,14 @@ For production:
 
 ## Examples
 
-See `examples/openrouter_vision_usage.rb` for complete working examples.
+- `examples/openrouter_vision_usage.rb` - Complete OpenRouter vision examples
+- `examples/zai_usage.rb` - Complete Z.ai GLM-4.5V examples including vision and text
 
 ## Further Reading
 
 - [OpenRouter Documentation](https://openrouter.ai/docs)
 - [OpenAI Vision API Reference](https://platform.openai.com/docs/guides/vision)
 - [Anthropic Claude Vision](https://docs.anthropic.com/claude/docs/vision)
+- [Z.ai API Platform](https://api.z.ai/)
+- [GLM-4.5V Documentation](https://bigmodel.cn/)
 

data/examples/zai_usage.rb

@@ -0,0 +1,163 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+# Example of Z.ai GLM model usage including multimodal/vision capabilities
+require_relative '../lib/llm_conductor'
+
+# Configure Z.ai
+LlmConductor.configure do |config|
+  config.zai(
+    api_key: ENV['ZAI_API_KEY']
+  )
+end
+
+# Example 1: Simple text-only request with GLM-4-plus
+puts '=== Example 1: Text-only request with GLM-4-plus ==='
+response = LlmConductor.generate(
+  model: 'glm-4-plus',
+  vendor: :zai,
+  prompt: 'What is the capital of France? Please answer in one sentence.'
+)
+puts response.output
+puts "Tokens used: #{response.total_tokens}\n\n"
+
+# Example 2: Text request with GLM-4.5V (vision model, text-only mode)
+puts '=== Example 2: Text-only request with GLM-4.5V ==='
+response = LlmConductor.generate(
+  model: 'glm-4.5v',
+  vendor: :zai,
+  prompt: 'Explain the concept of machine learning in simple terms.'
+)
+puts response.output
+puts "Tokens used: #{response.total_tokens}\n\n"
+
+# Example 3: Vision request with a single image
+puts '=== Example 3: Single image analysis with GLM-4.5V ==='
+response = LlmConductor.generate(
+  model: 'glm-4.5v',
+  vendor: :zai,
+  prompt: {
+    text: 'What do you see in this image? Please describe it in detail.',
+    images: 'https://upload.wikimedia.org/wikipedia/commons/thumb/d/dd/Gfp-wisconsin-madison-the-nature-boardwalk.jpg/2560px-Gfp-wisconsin-madison-the-nature-boardwalk.jpg'
+  }
+)
+puts response.output
+puts "Tokens used: #{response.total_tokens}\n\n"
+
+# Example 4: Vision request with multiple images
+puts '=== Example 4: Multiple images comparison with GLM-4.5V ==='
+response = LlmConductor.generate(
+  model: 'glm-4.5v',
+  vendor: :zai,
+  prompt: {
+    text: 'Compare these two images and describe the differences you observe.',
+    images: [
+      'https://upload.wikimedia.org/wikipedia/commons/thumb/d/dd/Gfp-wisconsin-madison-the-nature-boardwalk.jpg/2560px-Gfp-wisconsin-madison-the-nature-boardwalk.jpg',
+      'https://upload.wikimedia.org/wikipedia/commons/thumb/3/3f/Placeholder_view_vector.svg/681px-Placeholder_view_vector.svg.png'
+    ]
+  }
+)
+puts response.output
+puts "Tokens used: #{response.total_tokens}\n\n"
+
+# Example 5: Image with detail level specification
+puts '=== Example 5: Image with detail level ==='
+response = LlmConductor.generate(
+  model: 'glm-4.5v',
+  vendor: :zai,
+  prompt: {
+    text: 'Describe this image in detail, including colors, objects, and atmosphere.',
+    images: [
+      {
+        url: 'https://upload.wikimedia.org/wikipedia/commons/thumb/d/dd/Gfp-wisconsin-madison-the-nature-boardwalk.jpg/2560px-Gfp-wisconsin-madison-the-nature-boardwalk.jpg',
+        detail: 'high'
+      }
+    ]
+  }
+)
+puts response.output
+puts "Tokens used: #{response.total_tokens}\n\n"
+
+# Example 6: Using raw array format (advanced)
+puts '=== Example 6: Raw array format ==='
+response = LlmConductor.generate(
+  model: 'glm-4.5v',
+  vendor: :zai,
+  prompt: [
+    { type: 'text', text: 'What objects can you identify in this image?' },
+    {
+      type: 'image_url',
+      image_url: {
+        url: 'https://upload.wikimedia.org/wikipedia/commons/thumb/d/dd/Gfp-wisconsin-madison-the-nature-boardwalk.jpg/2560px-Gfp-wisconsin-madison-the-nature-boardwalk.jpg'
+      }
+    }
+  ]
+)
+puts response.output
+puts "Tokens used: #{response.total_tokens}\n\n"
+
+# Example 7: Base64 encoded image (for local images)
+puts '=== Example 7: Using base64 encoded image ==='
+# NOTE: In real usage, you would read and encode a local file
+# require 'base64'
+# image_data = Base64.strict_encode64(File.read('path/to/image.jpg'))
+# image_url = "data:image/jpeg;base64,#{image_data}"
+
+# For this example, we'll use a URL
+response = LlmConductor.generate(
+  model: 'glm-4.5v',
+  vendor: :zai,
+  prompt: {
+    text: 'Analyze this image and extract any text you can see.',
+    images: 'https://upload.wikimedia.org/wikipedia/commons/thumb/d/dd/Gfp-wisconsin-madison-the-nature-boardwalk.jpg/2560px-Gfp-wisconsin-madison-the-nature-boardwalk.jpg'
+  }
+)
+puts response.output
+puts "Tokens used: #{response.total_tokens}\n\n"
+
+# Example 8: Error handling
+puts '=== Example 8: Error handling ==='
+begin
+  response = LlmConductor.generate(
+    model: 'glm-4.5v',
+    vendor: :zai,
+    prompt: {
+      text: 'Analyze this image',
+      images: 'invalid-url'
+    }
+  )
+
+  if response.success?
+    puts response.output
+  else
+    puts "Error: #{response.metadata[:error]}"
+  end
+rescue StandardError => e
+  puts "Exception: #{e.message}"
+end
+
+# Example 9: Document understanding (OCR)
+puts "\n=== Example 9: Document understanding ==="
+response = LlmConductor.generate(
+  model: 'glm-4.5v',
+  vendor: :zai,
+  prompt: {
+    text: 'Please read any text visible in this image and transcribe it.',
+    images: 'https://upload.wikimedia.org/wikipedia/commons/thumb/d/dd/Gfp-wisconsin-madison-the-nature-boardwalk.jpg/2560px-Gfp-wisconsin-madison-the-nature-boardwalk.jpg'
+  }
+)
+puts response.output
+puts "Tokens used: #{response.total_tokens}\n\n"
+
+# Example 10: Complex reasoning with image
+puts '=== Example 10: Complex reasoning with image ==='
+response = LlmConductor.generate(
+  model: 'glm-4.5v',
+  vendor: :zai,
+  prompt: {
+    text: 'Based on this image, what time of day do you think it is? Explain your reasoning.',
+    images: 'https://upload.wikimedia.org/wikipedia/commons/thumb/d/dd/Gfp-wisconsin-madison-the-nature-boardwalk.jpg/2560px-Gfp-wisconsin-madison-the-nature-boardwalk.jpg'
+  }
+)
+puts response.output
+puts "Tokens used: #{response.total_tokens}\n\n"

data/lib/llm_conductor/client_factory.rb

@@ -19,7 +19,8 @@ module LlmConductor
         ollama: Clients::OllamaClient,
         gemini: Clients::GeminiClient,
         google: Clients::GeminiClient,
-        groq: Clients::GroqClient
+        groq: Clients::GroqClient,
+        zai: Clients::ZaiClient
       }
 
       client_classes.fetch(vendor) do
@@ -35,6 +36,8 @@ module LlmConductor
         :openai
       when /^gemini/i
         :gemini
+      when /^glm/i
+        :zai
       when /^(llama|mixtral|gemma|qwen)/i
         :groq
       else

data/lib/llm_conductor/clients/zai_client.rb

@@ -0,0 +1,153 @@
+# frozen_string_literal: true
+
+module LlmConductor
+  module Clients
+    # Z.ai client implementation for accessing GLM models including GLM-4.5V
+    # Supports both text-only and multimodal (vision) requests
+    #
+    # Note: Z.ai uses OpenAI-compatible API format but with /v4/ path instead of /v1/
+    # We use Faraday directly instead of the ruby-openai gem to properly handle the API path
+    class ZaiClient < BaseClient
+      private
+
+      # Override token calculation to handle multimodal content
+      def calculate_tokens(content)
+        case content
+        when String
+          super(content)
+        when Hash
+          # For multimodal content, count tokens only for text part
+          # Note: This is an approximation as images have variable token counts
+          text = content[:text] || content['text'] || ''
+          super(text)
+        when Array
+          # For pre-formatted arrays, extract and count text parts
+          text_parts = content.select { |part| part[:type] == 'text' || part['type'] == 'text' }
+                              .map { |part| part[:text] || part['text'] || '' }
+                              .join(' ')
+          super(text_parts)
+        else
+          super(content.to_s)
+        end
+      end
+
+      def generate_content(prompt)
+        content = format_content(prompt)
+
+        # Retry logic for transient errors (similar to OpenRouter)
+        max_retries = 3
+        retry_count = 0
+
+        begin
+          # Make direct HTTP request to Z.ai API since they use /v4/ instead of /v1/
+          response = http_client.post('chat/completions') do |req|
+            req.body = {
+              model:,
+              messages: [{ role: 'user', content: }]
+            }.to_json
+          end
+
+          # Response body is already parsed as Hash by Faraday's JSON middleware
+          response_data = response.body.is_a?(String) ? JSON.parse(response.body) : response.body
+          response_data.dig('choices', 0, 'message', 'content')
+        rescue Faraday::ServerError => e
+          retry_count += 1
+
+          # Log retry attempts if logger is configured
+          configuration.logger&.warn(
+            "Z.ai API error (attempt #{retry_count}/#{max_retries}): #{e.message}"
+          )
+
+          raise unless retry_count < max_retries
+
+          wait_time = 2**retry_count # Exponential backoff: 2, 4, 8 seconds
+          configuration.logger&.info("Retrying in #{wait_time}s...")
+          sleep(wait_time)
+          retry
+        end
+      end
+
+      # Format content based on whether it's a simple string or multimodal content
+      # @param prompt [String, Hash, Array] The prompt content
+      # @return [String, Array] Formatted content for the API
+      def format_content(prompt)
+        case prompt
+        when Hash
+          # Handle hash with text and/or images
+          format_multimodal_hash(prompt)
+        when Array
+          # Already formatted as array of content parts
+          prompt
+        else
+          # Simple string prompt
+          prompt.to_s
+        end
+      end
+
+      # Format a hash containing text and/or images into multimodal content array
+      # @param prompt_hash [Hash] Hash with :text and/or :images keys
+      # @return [Array] Array of content parts for the API
+      def format_multimodal_hash(prompt_hash)
+        content_parts = []
+
+        # Add text part if present
+        if prompt_hash[:text] || prompt_hash['text']
+          text = prompt_hash[:text] || prompt_hash['text']
+          content_parts << { type: 'text', text: }
+        end
+
+        # Add image parts if present
+        images = prompt_hash[:images] || prompt_hash['images'] || []
+        images = [images] unless images.is_a?(Array)
+
+        images.each do |image|
+          content_parts << format_image_part(image)
+        end
+
+        content_parts
+      end
+
+      # Format an image into the appropriate API structure
+      # @param image [String, Hash] Image URL or hash with url/detail keys
+      # @return [Hash] Formatted image part for the API
+      def format_image_part(image)
+        case image
+        when String
+          # Simple URL string or base64 data
+          { type: 'image_url', image_url: { url: image } }
+        when Hash
+          # Hash with url and optional detail level
+          {
+            type: 'image_url',
+            image_url: {
+              url: image[:url] || image['url'],
+              detail: image[:detail] || image['detail']
+            }.compact
+          }
+        end
+      end
+
+      # HTTP client for making requests to Z.ai API
+      # Z.ai uses /v4/ in their path, not /v1/ like OpenAI, so we use Faraday directly
+      def http_client
+        @http_client ||= begin
+          config = LlmConductor.configuration.provider_config(:zai)
+          base_url = config[:uri_base] || 'https://api.z.ai/api/paas/v4'
+
+          Faraday.new(url: base_url) do |f|
+            f.request :json
+            f.response :json
+            f.headers['Authorization'] = "Bearer #{config[:api_key]}"
+            f.headers['Content-Type'] = 'application/json'
+            f.adapter Faraday.default_adapter
+          end
+        end
+      end
+
+      # Legacy client method for compatibility (not used, but kept for reference)
+      def client
+        http_client
+      end
+    end
+  end
+end

data/lib/llm_conductor/configuration.rb

@@ -72,6 +72,14 @@ module LlmConductor
       }
     end
 
+    # Configure Z.ai provider
+    def zai(api_key: nil, **options)
+      @providers[:zai] = {
+        api_key: api_key || ENV['ZAI_API_KEY'],
+        **options
+      }
+    end
+
     # Get provider configuration
     def provider_config(provider)
       @providers[provider.to_sym] || {}
@@ -126,6 +134,14 @@ module LlmConductor
       groq(api_key: value)
     end
 
+    def zai_api_key
+      provider_config(:zai)[:api_key]
+    end
+
+    def zai_api_key=(value)
+      zai(api_key: value)
+    end
+
     private
 
     def setup_defaults_from_env
@@ -135,6 +151,7 @@ module LlmConductor
       openrouter if ENV['OPENROUTER_API_KEY']
       gemini if ENV['GEMINI_API_KEY']
       groq if ENV['GROQ_API_KEY']
+      zai if ENV['ZAI_API_KEY']
       ollama # Always configure Ollama with default URL
     end
   end

data/lib/llm_conductor/prompt_manager.rb

@@ -59,9 +59,7 @@ module LlmConductor
       def validate_prompt_class!(prompt_class)
         raise InvalidPromptClassError, 'Prompt must be a class' unless prompt_class.is_a?(Class)
 
-        unless prompt_class < Prompts::BasePrompt
-          raise InvalidPromptClassError, 'Prompt class must inherit from BasePrompt'
-        end
+        raise InvalidPromptClassError, 'Prompt class must inherit from BasePrompt' unless prompt_class < Prompts::BasePrompt
 
         return if prompt_class.instance_methods(false).include?(:render)
 

data/lib/llm_conductor/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module LlmConductor
-  VERSION = '1.1.2'
+  VERSION = '1.2.0'
 end

data/lib/llm_conductor.rb

@@ -14,10 +14,11 @@ require_relative 'llm_conductor/clients/groq_client'
 require_relative 'llm_conductor/clients/ollama_client'
 require_relative 'llm_conductor/clients/openrouter_client'
 require_relative 'llm_conductor/clients/gemini_client'
+require_relative 'llm_conductor/clients/zai_client'
 require_relative 'llm_conductor/client_factory'
 
 # LLM Conductor provides a unified interface for multiple Language Model providers
-# including OpenAI GPT, Anthropic Claude, Google Gemini, Groq, OpenRouter, and Ollama
+# including OpenAI GPT, Anthropic Claude, Google Gemini, Groq, OpenRouter, Z.ai, and Ollama
 # with built-in prompt templates, token counting, and extensible client architecture.
 module LlmConductor
   class Error < StandardError; end
@@ -63,16 +64,17 @@ module LlmConductor
       when :ollama then Clients::OllamaClient
       when :gemini, :google then Clients::GeminiClient
       when :groq then Clients::GroqClient
+      when :zai then Clients::ZaiClient
       else
         raise ArgumentError,
               "Unsupported vendor: #{vendor}. " \
-              'Supported vendors: anthropic, openai, openrouter, ollama, gemini, groq'
+              'Supported vendors: anthropic, openai, openrouter, ollama, gemini, groq, zai'
       end
     end
   end
 
   # List of supported vendors
-  SUPPORTED_VENDORS = %i[anthropic openai openrouter ollama gemini groq].freeze
+  SUPPORTED_VENDORS = %i[anthropic openai openrouter ollama gemini groq zai].freeze
 
   # List of supported prompt types
   SUPPORTED_PROMPT_TYPES = %i[

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: llm_conductor
 version: !ruby/object:Gem::Version
-  version: 1.1.2
+  version: 1.2.0
 platform: ruby
 authors:
 - Ben Zheng
@@ -161,6 +161,7 @@ files:
 - examples/prompt_registration.rb
 - examples/rag_usage.rb
 - examples/simple_usage.rb
+- examples/zai_usage.rb
 - lib/llm_conductor.rb
 - lib/llm_conductor/client_factory.rb
 - lib/llm_conductor/clients/anthropic_client.rb
@@ -170,6 +171,7 @@ files:
 - lib/llm_conductor/clients/groq_client.rb
 - lib/llm_conductor/clients/ollama_client.rb
 - lib/llm_conductor/clients/openrouter_client.rb
+- lib/llm_conductor/clients/zai_client.rb
 - lib/llm_conductor/configuration.rb
 - lib/llm_conductor/data_builder.rb
 - lib/llm_conductor/prompt_manager.rb