fast-deepseek 0.1.3 → 0.2.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 149d4a1cdffdafa982a1fffa3d77cbe2ec1636e8b9a2419521d6992b3132b963
-  data.tar.gz: 9ec6fd7ada1e529db50ea243cc543f661ce77d9e4f63b4667f99a92be44e2b68
+  metadata.gz: db93a692cecf12040497bbd26b9654b36a25f5b30887178dc6250016cd7a3cfe
+  data.tar.gz: 0add12910cf7304d4d778d1850cc1a09394377d2a04f5ec52b896e4f6f7cbb62
 SHA512:
-  metadata.gz: d4f2de1d54dff15fcc04f74b338c078463b7db6155650da61cd848c3539739fea44df707d52b9d5e3290626dffb786bbdf9d263e464e1dec058b75d85c84f5c4
-  data.tar.gz: a5f452e3c43bdf1425b7b02ad9cbc626b5c593fd7068225173e85ba7340a7f1636b56736d633007a3f6de7d5cce634d584a4750165b03ba6acb69f45f8e645c0
+  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.3".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,14 +1,28 @@
 --- !ruby/object:Gem::Specification
 name: fast-deepseek
 version: !ruby/object:Gem::Version
-  version: 0.1.3
+  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: dotenv
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.8'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.8'
 - !ruby/object:Gem::Dependency
   name: faraday
   requirement: !ruby/object:Gem::Requirement
@@ -34,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:
@@ -41,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/udaykadaboina/fast-deepseek/releases/tag/v0.1.3
+  changelog_uri: https://github.com/udaykadaboina/fast-deepseek/releases/tag/v0.2.2
   rubygems_mfa_required: 'true'
 rdoc_options: []
 require_paths:
@@ -50,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: []