fast-deepseek 0.1.2 → 0.2.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 3d6f10eeee14cd293415f0ee509aa3b84635e8e8597fc91b60d9726ceae0ab3a
-  data.tar.gz: 62cff10da5694252414b4360282e82b6d8dd2a1e6c3a3988f2d5eddd9f15ad32
+  metadata.gz: db93a692cecf12040497bbd26b9654b36a25f5b30887178dc6250016cd7a3cfe
+  data.tar.gz: 0add12910cf7304d4d778d1850cc1a09394377d2a04f5ec52b896e4f6f7cbb62
 SHA512:
-  metadata.gz: 6622e63258fff0761a275e51cf2f3c34aed8e32633d4c0a6697616609c1d3cfd343cc186f4446127b5ede2eefaac0e14c9042d59ab7e90a32e00d379c529c78e
-  data.tar.gz: e3c9e64d093d60ce2c55bedf4b0f1036f304d0214afe48ee98ce8278e32b6959eab2b35c584daa582f9ba5a95817ee5a1a88282c31186ad3e325b285b2428298
+  metadata.gz: c5c8594cd3f42a933aa85bd867ed9b55f74d08a0b8d4c8f926f423f5d52d9698e4b57377fbef3e88be5b7ba32f850ab5300f30956f9522686116cd46ec4a5343
+  data.tar.gz: d364afefc8880147b8ec64a22f50102cdfbe236770bcd481059b0e15934e3225d73232fd2db976e3d066b5e339a11ad86837012d267e25bff20d2954e6b31b26

data/lib/fast_deepseek/client.rb

@@ -1,60 +1,68 @@
-# lib/fast_deepseek/client.rb
-require 'faraday'
-require 'json'
-require 'logger'
-require 'dotenv/load'
+# frozen_string_literal: true
+
+require "faraday"
+require "json"
+require "logger"
+require "dotenv/load"
 
 module FastDeepseek
+  # Client for interacting with the DeepSeek API.
   class Client
-    DEFAULT_HOST = 'http://localhost:11434/api'.freeze
+    DEFAULT_HOST = "https://api.deepseek.com"
 
     def initialize(api_key: nil, base_url: DEFAULT_HOST)
-      @api_key = api_key || ENV['DEEPSEEK_API_KEY']
-      raise "DEEPSEEK_API_KEY is missing! Set it in the .env file or pass it explicitly." unless @api_key
+      @api_key = api_key || ENV.fetch("DEEPSEEK_API_KEY", nil)
+      raise Error, "DEEPSEEK_API_KEY is missing! Set it in the .env file or pass it explicitly." unless @api_key
 
-      @base_url = base_url
       @logger = Logger.new($stdout)
-
       @conn = Faraday.new(url: base_url) do |faraday|
         faraday.adapter Faraday.default_adapter
-        faraday.headers['Content-Type'] = 'application/json'
+        faraday.headers["Content-Type"] = "application/json"
       end
     end
 
-    def chat(prompt, model: , options: {})
-      request('chat', { model: model, messages: [{ role: 'user', content: prompt }], stream: false }.merge(options))
+    def chat(prompt, model:, options: {})
+      request("chat", { model: model, messages: [{ role: "user", content: prompt }], stream: false }.merge(options))
     end
 
-    private
-
-    def request(endpoint, payload)
-      response = @conn.post("/api/#{endpoint}", payload.to_json)
+    def models
+      request("models", method: :get)
+    end
 
-      response = @conn.post(endpoint) do |req|
-        req.headers['Authorization'] = "Bearer #{@api_key}"
-        req.body = JSON.dump(payload)
-      end
+    private
 
+    def request(endpoint, payload = nil, method: :post)
+      response = send_request(endpoint, payload, method)
       handle_response(response)
     rescue Faraday::Error => e
       @logger.error("API request failed: #{e.message}")
-      raise DeepSeek::Error, "API request failed: #{e.message}"
-      JSON.parse(response.body)
+      raise Error, "API request failed: #{e.message}"
+    rescue JSON::ParserError => e
+      @logger.error("Invalid API response: #{e.message}")
+      raise Error, "Invalid API response: #{e.message}"
+    end
+
+    def send_request(endpoint, payload, method)
+      path = method == :post ? "#{endpoint}/completions" : endpoint
+      @conn.public_send(method, path) do |req|
+        req.headers["Authorization"] = format("Bearer %s", @api_key)
+        req.body = payload.to_json if payload
+      end
     end
 
     def handle_response(response)
       case response.status
       when 429
-        @logger.error('API rate limit exceeded. Please try again later.')
-        raise DeepSeek::RateLimitError, 'API rate limit exceeded. Please try again later.'
+        @logger.error("API rate limit exceeded. Please try again later.")
+        raise RateLimitError, "API rate limit exceeded. Please try again later."
       when 500..599
-        @logger.error('Server error occurred. Please try again later.')
-        raise DeepSeek::ServerError, 'Server error occurred. Please try again later.'
-      when 200
+        @logger.error("Server error occurred. Please try again later.")
+        raise ServerError, "Server error occurred. Please try again later."
+      when 200..299
         JSON.parse(response.body)
       else
         @logger.error("API request failed: #{response.status} - #{response.body}")
-        raise DeepSeek::Error, "API request failed: #{response.status} - #{response.body}"
+        raise Error, "API request failed: #{response.status} - #{response.body}"
       end
     end
   end

data/lib/fast_deepseek/errors.rb

@@ -0,0 +1,9 @@
+# frozen_string_literal: true
+
+module FastDeepseek
+  class Error < StandardError; end
+
+  class RateLimitError < Error; end
+
+  class ServerError < Error; end
+end

data/lib/fast_deepseek/version.rb

@@ -1,4 +1,6 @@
+# frozen_string_literal: true
+
 # lib/fast_deepseek/version.rb
 module FastDeepseek
-  VERSION = "0.1.2".freeze
+  VERSION = "0.2.2"
 end

data/lib/fast_deepseek.rb

@@ -1,3 +1,5 @@
-# lib/fast_deepseek.rb
+# frozen_string_literal: true
+
 require_relative "fast_deepseek/version"
+require_relative "fast_deepseek/errors"
 require_relative "fast_deepseek/client"

metadata

@@ -1,56 +1,42 @@
 --- !ruby/object:Gem::Specification
 name: fast-deepseek
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.2.2
 platform: ruby
 authors:
 - Uday Kadaboina
 bindir: exe
 cert_chain: []
-date: 2025-03-26 00:00:00.000000000 Z
+date: 1980-01-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
-  name: faraday
+  name: dotenv
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '2.0'
+        version: '2.8'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '2.0'
+        version: '2.8'
 - !ruby/object:Gem::Dependency
-  name: rspec
-  requirement: !ruby/object:Gem::Requirement
-    requirements:
-    - - "~>"
-      - !ruby/object:Gem::Version
-        version: '3.10'
-  type: :development
-  prerelease: false
-  version_requirements: !ruby/object:Gem::Requirement
-    requirements:
-    - - "~>"
-      - !ruby/object:Gem::Version
-        version: '3.10'
-- !ruby/object:Gem::Dependency
-  name: webmock
+  name: faraday
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '3.14'
-  type: :development
+        version: '2.0'
+  type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '3.14'
+        version: '2.0'
 description: Provides an easy-to-use interface for interacting with DeepSeek API.
 email:
 - uday88k@gmail.com
@@ -62,6 +48,7 @@ files:
 - bin/setup
 - lib/fast_deepseek.rb
 - lib/fast_deepseek/client.rb
+- lib/fast_deepseek/errors.rb
 - lib/fast_deepseek/version.rb
 homepage: https://github.com/udaykadaboina/fast-deepseek
 licenses:
@@ -69,7 +56,7 @@ licenses:
 metadata:
   homepage_uri: https://github.com/udaykadaboina/fast-deepseek
   source_code_uri: https://github.com/udaykadaboina/fast-deepseek
-  changelog_uri: https://github.com/rails/rails/releases/tag/v0.1.2
+  changelog_uri: https://github.com/udaykadaboina/fast-deepseek/releases/tag/v0.2.2
   rubygems_mfa_required: 'true'
 rdoc_options: []
 require_paths:
@@ -78,14 +65,14 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: 2.6.0
+      version: 3.2.0
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.6.6
+rubygems_version: 4.0.21
 specification_version: 4
 summary: A fast wrapper for DeepSeek API
 test_files: []