fast-deepseek 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: '029e9f8f90a06915a4bc2a08f1947dcd340b52414f9696927e10f5efefc9d41d'
+  data.tar.gz: fc0f90df9a49796a4a158d6bca50cf69d6b44dc1a14112b224af0aff27c17763
+SHA512:
+  metadata.gz: 8c40350a4f3e8bba881c722441b92d1b3f0c1fb5834175963a12e57322d945dca7811c446227c69eb9b537bebc24956b14eeffd606917191fc6c33e442f9fa2d
+  data.tar.gz: 7a22bd10dfa54c8e8b5ca2a940c0fba3b062e5f94214a040e48c039daee99b8f1f06bd59ffd84365d48bbb93157dd1646f0a34413645aea2bf7051d47a7e2608

data/bin/console

@@ -0,0 +1,15 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+require "bundler/setup"
+require "fast_deepseek"
+
+# You can add fixtures and/or initialization code here to make experimenting
+# with your gem easier. You can also use a different console, if you like.
+
+# (If you use this, don't forget to add pry to your Gemfile!)
+# require "pry"
+# Pry.start
+
+require "irb"
+IRB.start(__FILE__)

data/bin/setup

@@ -0,0 +1,8 @@
+#!/usr/bin/env bash
+set -euo pipefail
+IFS=$'\n\t'
+set -vx
+
+bundle install
+
+# Do any other automated setup that you need to do here

data/lib/fast_deepseek/client.rb

@@ -0,0 +1,62 @@
+# lib/fast_deepseek/client.rb
+require 'faraday'
+require 'json'
+require 'logger'
+
+module FastDeepseek
+  class Client
+    DEFAULT_HOST = 'http://localhost:11434/api'.freeze
+
+    def initialize(api_key:, base_url: DEFAULT_HOST)
+      @api_key = api_key || ENV['DEEPSEEK_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'
+      end
+    end
+
+    def chat(prompt, model: 'deepseek-r1:1.5b', options: {})
+      request('chat', { model: model, messages: [{ role: 'user', content: prompt }], stream: false }.merge(options))
+    end
+
+    def coder(prompt, model: 'deepseek-coder')
+      request('chat', { model: model, messages: [{ role: 'user', content: prompt }], stream: false })
+    end
+
+    private
+
+    def request(endpoint, payload)
+      response = @conn.post("/api/#{endpoint}", payload.to_json)
+
+      response = @conn.post(endpoint) do |req|
+        req.headers['Authorization'] = "Bearer #{@api_key}"
+        req.body = JSON.dump(payload)
+      end
+
+      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)
+    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.'
+      when 500..599
+        @logger.error('Server error occurred. Please try again later.')
+        raise DeepSeek::ServerError, 'Server error occurred. Please try again later.'
+      when 200
+        JSON.parse(response.body)
+      else
+        @logger.error("API request failed: #{response.status} - #{response.body}")
+        raise DeepSeek::Error, "API request failed: #{response.status} - #{response.body}"
+      end
+    end
+  end
+end

data/lib/fast_deepseek/version.rb

@@ -0,0 +1,4 @@
+# lib/fast_deepseek/version.rb
+module FastDeepseek
+  VERSION = '0.1.0'.freeze
+end

data/lib/fast_deepseek.rb

@@ -0,0 +1,3 @@
+# lib/fast_deepseek.rb
+require_relative "fast_deepseek/version"
+require_relative "fast_deepseek/client"

metadata

@@ -0,0 +1,93 @@
+--- !ruby/object:Gem::Specification
+name: fast-deepseek
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Uday Kadaboina
+autorequire:
+bindir: exe
+cert_chain: []
+date: 2025-02-21 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: faraday
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+- !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
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.14'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.14'
+description: Provides an easy-to-use interface for interacting with DeepSeek API.
+email:
+- uday88k@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- bin/console
+- bin/setup
+- lib/fast_deepseek.rb
+- lib/fast_deepseek/client.rb
+- lib/fast_deepseek/version.rb
+homepage: https://github.com/udaykadaboina/fast-deepseek
+licenses:
+- MIT
+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/CHANGELOG.md
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 2.6.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.5.11
+signing_key:
+specification_version: 4
+summary: A fast wrapper for DeepSeek API
+test_files: []