geekdict 0.1.1 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f9a7876ba4eddfb0497523ab7f318588fc1ef1abf754b5f75771b0f2bbc29092
-  data.tar.gz: 460d3c363a9622fec12bb463854239355293827327d7d7639338c306a36f99a3
+  metadata.gz: 8c60ec472c37345eac7c1b4be1f60c56ea5edca02c3c5af8eb19be3cd1173470
+  data.tar.gz: ada5e008be1245e52e1b70827b7ef517f067ce187a3a7c755d97e45c189cb14d
 SHA512:
-  metadata.gz: a66b0934455eebc3bc5c511510abc261d7c8ddb59cc91f4ea66b4b30ee6b8d709f39451e62d4674731db803ec88f105c064c8722b56b9e073abd0c38597b8c1a
-  data.tar.gz: 64d730010c52a0e5c7a98b169ff7fef55ed34eab4863b32781aefd18889ffa946cab96dbc5d33516668488d87ddb3be830a111a7b75092e552c3fc29f2d106f2
+  metadata.gz: 06ae16f269a32062b8d825debf6e4f76df6cf9e65394365c8275dd0846d163bcaa70203d39561695c6744ae9e137364d6cd2b2ce4d2ee9bb59df492bed357022
+  data.tar.gz: e29388ec6089b5172b6898f805942150c8862e0b909ba0dbc37ee4e2d3835b89d190128e6824b19bef7905174dcd8b1c7654f6b3debc176d9e72536fc1be100f

data/README.md

@@ -12,38 +12,130 @@ You can install via rubygems:
 
 Configure
 --------
-This translate uses OpenAI's GPT-3.5. Ensure you have `OPENAI_API_KEY` configured as environment variable.
+GeekDict requires API keys for certain providers, set as environment variables:
+
+*   **OpenRouter**: Set `OPENROUTER_API_KEY`.
+*   **OpenAI**: Set `OPENAI_API_KEY`.
+*   **Youdao**: Configuration for Youdao keys is typically handled within its specific module or a separate config file if needed (refer to Youdao provider specifics if used).
+
+### Configuration File (Optional)
+
+You can customize the default provider and model by creating a configuration file at `~/.geekdict.config`. The file should be in YAML format.
+
+**Example `~/.geekdict.config`:**
+
+```yaml
+provider: openrouter # 'openai', 'openrouter', or 'youdao'
+model: google/gemini-2.5-flash-preview # Specific model for the chosen provider
+```
+
+**Defaults:**
+
+*   If the config file is not present or a setting is missing, the tool defaults to:
+    *   `provider: openrouter`
+    *   `model: google/gemini-2.5-flash-preview`
+
+**Prioritization:**
+
+Settings are applied in the following order (highest priority first):
+
+1.  Command-line options (`--provider`, `--model`)
+2.  Values in `~/.geekdict.config`
+3.  Hardcoded defaults
 
 
 Commands
 --------
-### Translate a word
+### Translate a word (`t`)
+
+Translates a given word between English and Chinese.
+
+**Usage:**
+
+```bash
+geekdict t [options] <word>
+```
+
+**Options:**
+
+*   `-p, --provider PROVIDER`: Specify the translation provider (`openai`, `openrouter`, `youdao`). Overrides config file setting.
+*   `-m, --model MODEL`: Specify the LLM model to use (e.g., `gpt-4`, `google/gemini-pro`). Overrides config file setting.
+*   `-d, --debug`: Enable debug output.
+*   `-o, --open`: (Functionality might vary - check specific provider usage)
+
+**Examples:**
 
-	$geekdict t test
-	n. 试验；检验
-	vt. 试验；测试
-	vi. 试验；测试
-	n. (Test)人名；(英)特斯特
+```bash
+# Use defaults (provider/model from CLI -> config -> hardcoded defaults)
+$ geekdict t hello
+
+# Specify provider (uses default/config model for that provider)
+$ geekdict t hello -p openai
+
+# Specify model (uses default/config provider)
+$ geekdict t hello -m gpt-4
+
+# Specify both provider and model
+$ geekdict t hello --provider openrouter --model anthropic/claude-3-haiku
+
+# Use Youdao provider (model option typically ignored)
+$ geekdict t hello -p youdao
+```
+
+**Example Output (using an LLM provider):**
+
+```
+Translation: 你好 (nǐ hǎo)
+
+Explanation:
+"你好" (nǐ hǎo) is the standard Mandarin Chinese greeting equivalent to "hello" in English. It's a polite and common way to greet someone.
+
+Examples:
+1. 你好，请问有什么可以帮您的吗？
+   Hello, may I help you with anything?
+2. 他笑着对我说你好。
+   He smiled and said hello to me.
+```
 
 Command Help
 ------------
-Use *help* command to get detail information.
+Use the `help` command to get detailed information about commands and options.
 
-	$geekdict
-	Commands:
-	  geekdict help [COMMAND]  # Describe available commands or one specific command
-	  geekdict t               # Translate a word
+```bash
+$ geekdict help
+Commands:
+  geekdict help [COMMAND]  # Describe available commands or one specific command
+  geekdict t <word>        # Translate a word
+  geekdict v               # Show version
 
-	$geekdict help t
-	Usage:
-	  geekdict t
+$ geekdict help t
+Usage:
+  geekdict t [options] <word>
 
-	Options:
+Options:
   -d, [--debug], [--no-debug]
   -o, [--open], [--no-open]
+  -p, [--provider=PROVIDER]              # Provider (overrides config: openai/openrouter/youdao)
+  -m, [--model=MODEL]                    # LLM model (overrides config)
+                                         # Default: false
+
+Translate a word
+```
+
+### Shell Alias (Optional)
 
-	Translate a word
+For faster access, you can create a shell alias. Add the following line to your shell configuration file (e.g., `~/.bashrc`, `~/.zshrc`):
 
+```bash
+alias t='geekdict t'
+```
+
+After adding the alias and restarting your shell or sourcing the configuration file (e.g., `source ~/.zshrc`), you can simply use `t` instead of `geekdict t`:
+
+```bash
+# Instead of: geekdict t hello
+t hello
+```
 
 Development
 ----------

data/lib/geekdict/cli.rb

@@ -1,21 +1,69 @@
 require "thor"
+# Remove YAML require, handled in config.rb
 require_relative 'local_history'
 require_relative 'version'
+require_relative 'config' # Add require for the new config module
 require_relative 'openai/gpt.rb'
+require_relative 'openrouter/api.rb'
+# Assuming youdao might be added later or is handled elsewhere
+# require_relative 'youdao/api.rb'
+require_relative 'youdao/api.rb' # Keep youdao require
 
 module GeekDict
+  class CLI < Thor
+    # Constants and load_config are now in GeekDict::Config
 
-    class CLI < Thor
+    desc "t", "Translate a word"
+    option :debug, :aliases => '-d', :type => :boolean, :default => false
+    option :open, :aliases => '-o', :type => :boolean, :default => false
+    # Update enum and desc to use Config constants
+    option :provider, :aliases => '-p', :type => :string, :enum => GeekDict::Config::ALLOWED_PROVIDERS, :desc => "Provider (overrides config: #{GeekDict::Config::ALLOWED_PROVIDERS.join('/')})"
+    # Add model option
+    option :model, :aliases => '-m', :type => :string, :desc => "LLM model (overrides config)"
+    def t(word)
+      GeekDict.debugger options[:debug]
+      LocalHistory.save word
 
-        desc "t", "Translate a word"
-        option :debug, :aliases=>'-d', :type=>:boolean,:default=>false
-        option :open, :aliases => '-o', :type => :boolean, :default=> false
-        def t(word)
-            GeekDict.debugger options[:debug]
-            LocalHistory.save word
-            result = GeekDict::OpenAI.translate word
-            puts result
-        end
+      # 1. Load config file using the Config module
+      config = GeekDict::Config.load_config
+
+      # 2. Determine provider: CLI option > Config file > Default (use Config constants)
+      provider = options[:provider]&.downcase || config[:provider] || GeekDict::Config::DEFAULT_PROVIDER
+
+      # 3. Determine model: CLI option > Config file > Default (use Config constants)
+      model = options[:model] || config[:model] || GeekDict::Config::DEFAULT_MODEL
+
+      # 4. Validate provider (use Config constants)
+      unless GeekDict::Config::ALLOWED_PROVIDERS.include?(provider)
+        warn "Warning: Invalid provider '#{provider}' specified. Using default '#{GeekDict::Config::DEFAULT_PROVIDER}'."
+        provider = GeekDict::Config::DEFAULT_PROVIDER
+        # Reset model if provider changed to default, unless model was explicitly set via CLI
+        model = options[:model] || GeekDict::Config::DEFAULT_MODEL # Use default model for the default provider
+      end
+
+      # Optional: Output effective settings for debugging/clarity
+      puts "Using provider: #{provider}, model: #{model}" if options[:debug]
+
+      # 5. Call the appropriate provider with the word and model
+      result = case provider
+               when 'openai'
+                 # TODO: Update OpenAI class to accept model
+                 GeekDict::OpenAI.translate(word, model: model)
+               when 'openrouter'
+                 # TODO: Update OpenRouter class to accept model
+                 GeekDict::OpenRouter.translate(word, model: model)
+               when 'youdao'
+                 # Youdao might not use a 'model' in the same way.
+                 # If it needs config, adjust here. For now, pass nothing extra.
+                 GeekDict::Youdao.translate(word)
+               else
+                 # This case should technically not be reached due to validation
+                 warn "Internal Error: Unknown provider '#{provider}'. Aborting."
+                 exit 1
+               end
+
+      puts result
+    end
 
         desc "v", "version"
         def v()

data/lib/geekdict/config.rb

@@ -0,0 +1,33 @@
+require "yaml"
+
+module GeekDict
+  module Config
+    CONFIG_PATH = File.expand_path("~/.geekdict.config")
+    DEFAULT_PROVIDER = "openrouter"
+    DEFAULT_MODEL = "google/gemini-2.5-flash-preview"
+    ALLOWED_PROVIDERS = ['openai', 'openrouter', 'youdao'].freeze
+
+    module_function
+
+    # Method to load configuration from ~/.geekdict.config
+    def load_config
+      config = {}
+      if File.exist?(CONFIG_PATH)
+        begin
+          loaded_config = YAML.load_file(CONFIG_PATH)
+          config = loaded_config if loaded_config.is_a?(Hash)
+        rescue Psych::SyntaxError => e
+          warn "Warning: Error parsing config file #{CONFIG_PATH}: #{e.message}. Using defaults."
+        rescue => e
+          warn "Warning: Could not load config file #{CONFIG_PATH}: #{e.message}. Using defaults."
+        end
+      end
+      # Ensure keys are symbols for easier access, handle potential nil values
+      config = config.transform_keys { |k| k.to_s.downcase.to_sym rescue k } # Make keys symbols & lowercase
+      {
+        provider: config[:provider],
+        model: config[:model]
+      }
+    end
+  end
+end

data/lib/geekdict/openai/gpt.rb

@@ -8,12 +8,16 @@ module GeekDict
 
 		module_function
 
-		def translate(word)
+    # Update translate method to accept model
+		def translate(word, model: nil) # Add model keyword argument
 			@debugger = GeekDict.debugger
+      # Use the provided model, or fallback to a default if nil
+      effective_model = model || "gpt-3.5-turbo" # Fallback model for OpenAI
+
 			client = ::OpenAI::Client.new(access_token: ENV.fetch('OPENAI_API_KEY'))
       response = client.chat(
         parameters: {
-          model: "gpt-3.5-turbo",
+          model: effective_model, # Use the determined model
           messages: [
             { role: "system", content: system_prompt(word)},
             { role: "user", content: word}
@@ -25,7 +29,24 @@ module GeekDict
 
     def system_prompt(word)
       <<-EOS
-      You are helpful and enthusiastic language translator. If the word or sentense is Chinese, translate it into English. If it is English, translate into Chinese. If the word is incorrectly spelled or it's not a correct word in English or Chinese, try your best to find the closest word. In addition to the translation, offer a comprehensive explanation of the word along with one or two examples to illustrate its usage. Do NOT add Pinyin in the example sentences for Chinese.
+      You are a precise language translator specializing in English-Chinese translation. Follow these guidelines:
+
+      1. If the input is Chinese, translate to English. If English, translate to Chinese.
+      2. For words with multiple meanings, prioritize the most common usage first.
+      3. For misspelled or incorrect words, suggest the closest correct word.
+      4. For idiomatic expressions, provide both literal and figurative translations.
+      5. For technical terms, include the field/domain where appropriate.
+
+      Format your response as follows:
+      - Translation: [primary translation]
+      - Explanation: [concise explanation of meaning and usage]
+      - Examples:
+        1. [example sentence in target language]
+           [translation in source language]
+        2. [second example if helpful]
+           [translation in source language]
+
+      Keep explanations clear and concise. Do NOT include pinyin in Chinese text.
       EOS
     end
 	end

data/lib/geekdict/openrouter/api.rb

@@ -0,0 +1,67 @@
+require 'json'
+require 'httpclient'
+
+module GeekDict
+  module OpenRouter
+    module_function
+
+    # Update translate method to accept model
+    def translate(word, model: nil) # Add model keyword argument
+      @debugger = GeekDict.debugger
+      
+      # Use the provided model, or fallback to a default if nil (though CLI should provide one)
+      effective_model = model || 'google/gemini-2.5-flash-preview' # Fallback, though CLI provides default
+
+      client = HTTPClient.new
+      headers = {
+        'Content-Type' => 'application/json',
+        'Authorization' => "Bearer #{ENV.fetch('OPENROUTER_API_KEY')}"
+      }
+      
+      body = {
+        model: effective_model, # Use the determined model
+        messages: [
+          { role: "system", content: system_prompt(word) },
+          { role: "user", content: word }
+        ],
+        temperature: 0.2
+      }
+      
+      response = client.post(
+        'https://openrouter.ai/api/v1/chat/completions',
+        body.to_json,
+        headers
+      )
+      
+      if response.status == 200
+        result = JSON.parse(response.body)
+        result.dig("choices", 0, "message", "content")
+      else
+        "Error: Failed to get translation (#{response.status})"
+      end
+    end
+
+    def system_prompt(word)
+      <<-EOS
+      You are a precise language translator specializing in English-Chinese translation. Follow these guidelines:
+
+      1. If the input is Chinese, translate to English. If English, translate to Chinese.
+      2. For words with multiple meanings, prioritize the most common usage first.
+      3. For misspelled or incorrect words, suggest the closest correct word.
+      4. For idiomatic expressions, provide both literal and figurative translations.
+      5. For technical terms, include the field/domain where appropriate.
+
+      Format your response as follows:
+      - Translation: [primary translation]
+      - Explanation: [concise explanation of meaning and usage]
+      - Examples:
+        1. [example sentence in target language]
+           [translation in source language]
+        2. [second example if helpful]
+           [translation in source language]
+
+      Keep explanations clear and concise. Do NOT include pinyin in Chinese text.
+      EOS
+    end
+  end
+end

data/lib/geekdict/version.rb

@@ -1,3 +1,3 @@
 module GeekDict
-  VERSION = "0.1.1"
+  VERSION = "0.2.0"
 end

data/lib/geekdict.rb

@@ -1,6 +1,8 @@
 require 'geekdict/cli'
 require 'geekdict/debugger'
 require 'geekdict/youdao/api'
+require 'geekdict/openai/gpt'
+require 'geekdict/openrouter/api'
 
 module GeekDict
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: geekdict
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.2.0
 platform: ruby
 authors:
 - Wenbing Li
-autorequire: 
+autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-05-01 00:00:00.000000000 Z
+date: 2025-04-22 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rake
@@ -110,9 +110,11 @@ files:
 - geekdict.gemspec
 - lib/geekdict.rb
 - lib/geekdict/cli.rb
+- lib/geekdict/config.rb
 - lib/geekdict/debugger.rb
 - lib/geekdict/local_history.rb
 - lib/geekdict/openai/gpt.rb
+- lib/geekdict/openrouter/api.rb
 - lib/geekdict/version.rb
 - lib/geekdict/youdao/api.rb
 - lib/geekdict/youdao/config.rb
@@ -123,7 +125,7 @@ homepage: https://github.com/wbinglee/geekdict
 licenses:
 - MIT
 metadata: {}
-post_install_message: 
+post_install_message:
 rdoc_options: []
 require_paths:
 - lib
@@ -138,8 +140,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.3.26
-signing_key: 
+rubygems_version: 3.5.22
+signing_key:
 specification_version: 4
 summary: A command line tool for translation.
 test_files: