prescient 0.4.0 → 0.5.0

data/schema/prescient.configuration.schema.json

@@ -0,0 +1,153 @@
+{
+  "$schema": "https://json-schema.org/draft/2020-12/schema",
+  "$id": "https://github.com/kanutocd/prescient/schema/prescient.configuration.schema.json",
+  "title": "Prescient Configuration",
+  "type": "object",
+  "additionalProperties": false,
+  "properties": {
+    "$schema": {
+      "type": "string"
+    },
+    "version": {
+      "const": 1
+    },
+    "default_provider": {
+      "type": "string"
+    },
+    "default_provider_env": {
+      "type": "string"
+    },
+    "timeout": {
+      "type": "number"
+    },
+    "timeout_env": {
+      "type": "string"
+    },
+    "retry_attempts": {
+      "type": "integer"
+    },
+    "retry_attempts_env": {
+      "type": "string"
+    },
+    "retry_delay": {
+      "type": "number"
+    },
+    "retry_delay_env": {
+      "type": "string"
+    },
+    "fallback_providers": {
+      "type": "array",
+      "items": {
+        "type": "string"
+      }
+    },
+    "fallback_providers_env": {
+      "type": "string"
+    },
+    "sensitive_keys": {
+      "type": "array",
+      "items": {
+        "type": "string"
+      }
+    },
+    "sensitive_keys_env": {
+      "type": "string"
+    },
+    "providers": {
+      "type": "object",
+      "additionalProperties": {
+        "$ref": "#/$defs/provider"
+      }
+    }
+  },
+  "$defs": {
+    "provider": {
+      "type": "object",
+      "additionalProperties": false,
+      "required": [
+        "type"
+      ],
+      "properties": {
+        "type": {
+          "type": "string",
+          "enum": [
+            "ollama",
+            "anthropic",
+            "openai",
+            "huggingface",
+            "gemini",
+            "mistral",
+            "deepseek",
+            "xai"
+          ]
+        },
+        "api_key": {
+          "type": "string"
+        },
+        "api_key_env": {
+          "type": "string"
+        },
+        "url": {
+          "type": "string"
+        },
+        "url_env": {
+          "type": "string"
+        },
+        "model": {
+          "type": "string"
+        },
+        "model_env": {
+          "type": "string"
+        },
+        "chat_model": {
+          "type": "string"
+        },
+        "chat_model_env": {
+          "type": "string"
+        },
+        "embedding_model": {
+          "type": "string"
+        },
+        "embedding_model_env": {
+          "type": "string"
+        },
+        "embedding_dimensions": {
+          "type": "integer"
+        },
+        "embedding_dimensions_env": {
+          "type": "string"
+        },
+        "timeout": {
+          "type": "number"
+        },
+        "timeout_env": {
+          "type": "string"
+        },
+        "context_configs": {
+          "type": "object"
+        },
+        "context_configs_env": {
+          "type": "string"
+        },
+        "prompt_templates": {
+          "type": "object",
+          "additionalProperties": false,
+          "properties": {
+            "system_prompt": {
+              "type": "string"
+            },
+            "no_context_template": {
+              "type": "string"
+            },
+            "with_context_template": {
+              "type": "string"
+            }
+          }
+        },
+        "prompt_templates_env": {
+          "type": "string"
+        }
+      }
+    }
+  }
+}

data/sig/prescient.rbs

@@ -154,6 +154,82 @@ module Prescient
       def validate_configuration!: () -> void
       def validate_response!: (untyped, String) -> void
     end
+
+    class Gemini < Base
+      def self.base_uri: (String) -> untyped
+      def self.default_timeout: (Integer) -> untyped
+      def self.get: (String, **untyped) -> untyped
+      def self.post: (String, **untyped) -> untyped
+
+      def initialize: (**untyped) -> void
+      def generate_embedding: (String, **untyped) -> Array[Float]
+      def generate_response: (String, ?Array[untyped], **untyped) -> Hash[Symbol, untyped]
+      def health_check: () -> Hash[Symbol, untyped]
+      def list_models: () -> Array[Hash[Symbol, untyped]]
+
+      private
+
+      def validate_configuration!: () -> void
+      def api_headers: () -> Hash[String, String]
+      def model_endpoint: (untyped, String) -> String
+      def find_model: (Array[untyped], untyped, String) -> bool
+    end
+
+    class Mistral < Base
+      def self.base_uri: (String) -> untyped
+      def self.default_timeout: (Integer) -> untyped
+      def self.get: (String, **untyped) -> untyped
+      def self.post: (String, **untyped) -> untyped
+
+      def initialize: (**untyped) -> void
+      def generate_embedding: (String, **untyped) -> Array[Float]
+      def generate_response: (String, ?Array[untyped], **untyped) -> Hash[Symbol, untyped]
+      def health_check: () -> Hash[Symbol, untyped]
+      def list_models: () -> Array[Hash[Symbol, untyped]]
+
+      private
+
+      def validate_configuration!: () -> void
+      def api_headers: () -> Hash[String, String]
+      def model_available?: (Array[untyped], untyped) -> bool
+      def normalize_content: (untyped) -> String?
+    end
+
+    class DeepSeek < Base
+      def self.base_uri: (String) -> untyped
+      def self.default_timeout: (Integer) -> untyped
+      def self.get: (String, **untyped) -> untyped
+      def self.post: (String, **untyped) -> untyped
+
+      def initialize: (**untyped) -> void
+      def generate_embedding: (String, **untyped) -> untyped
+      def generate_response: (String, ?Array[untyped], **untyped) -> Hash[Symbol, untyped]
+      def health_check: () -> Hash[Symbol, untyped]
+      def list_models: () -> Array[Hash[Symbol, untyped]]
+
+      private
+
+      def validate_configuration!: () -> void
+      def api_headers: () -> Hash[String, String]
+    end
+
+    class XAI < Base
+      def self.base_uri: (String) -> untyped
+      def self.default_timeout: (Integer) -> untyped
+      def self.get: (String, **untyped) -> untyped
+      def self.post: (String, **untyped) -> untyped
+
+      def initialize: (**untyped) -> void
+      def generate_embedding: (String, **untyped) -> untyped
+      def generate_response: (String, ?Array[untyped], **untyped) -> Hash[Symbol, untyped]
+      def health_check: () -> Hash[Symbol, untyped]
+      def list_models: () -> Array[Hash[Symbol, untyped]]
+
+      private
+
+      def validate_configuration!: () -> void
+      def api_headers: () -> Hash[String, String]
+    end
   end
 
   module Pgvector
@@ -222,9 +298,30 @@ module Prescient
     def run: () -> Integer
   end
 
+  class ConfigurationLoader
+    CONFIGURATION_VERSION: Integer
+    TOP_LEVEL_KEYS: Array[String]
+    PROVIDER_TYPES: Hash[String, untyped]
+    COMMON_PROVIDER_KEYS: Array[String]
+    PROMPT_TEMPLATE_KEYS: Array[String]
+    ATTR_KEYS: Array[String]
+
+    def self.load_file: (String, ?env: untyped) -> Configuration
+    def self.load_yaml: (String, ?env: untyped) -> Configuration
+    def self.load_hash: (Hash[untyped, untyped], ?env: untyped) -> Configuration
+    def self.schema_path: () -> String
+
+    def initialize: (?untyped) -> void
+    def load_file: (String) -> Configuration
+    def load_yaml: (String, ?source: String?) -> Configuration
+    def load_hash: (untyped, ?source: String?) -> Configuration
+    def apply!: (Configuration, untyped, ?source: String?) -> Configuration
+  end
+
   def self.configure: () { (Configuration) -> void } -> void
   def self.configuration: () -> Configuration
   def self.reset_configuration!: () -> Configuration
+  def self.load_configuration: (?String?, ?env: untyped) -> Configuration
   def self.client: (?Symbol, ?enable_fallback: bool, ?provider_options: Hash[Symbol, untyped]) -> Client
   def self.generate_embedding: (String, ?provider: Symbol, ?enable_fallback: bool, **untyped) -> Array[Float]
   def self.generate_response: (String, ?Array[untyped], ?provider: Symbol, ?enable_fallback: bool, **untyped) -> Hash[Symbol, untyped]
@@ -233,4 +330,12 @@ module Prescient
   private
 
   def self.configure_default_providers: (Configuration, untyped) -> void
+  def self.configure_ollama: (Configuration, untyped) -> void
+  def self.configure_openai: (Configuration, untyped) -> void
+  def self.configure_anthropic: (Configuration, untyped) -> void
+  def self.configure_gemini: (Configuration, untyped) -> void
+  def self.configure_mistral: (Configuration, untyped) -> void
+  def self.configure_deepseek: (Configuration, untyped) -> void
+  def self.configure_xai: (Configuration, untyped) -> void
+  def self.configure_huggingface: (Configuration, untyped) -> void
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: prescient
 version: !ruby/object:Gem::Version
-  version: 0.4.0
+  version: 0.5.0
 platform: ruby
 authors:
 - Ken C. Demanawa
@@ -61,13 +61,19 @@ files:
 - lib/prescient/base.rb
 - lib/prescient/cli.rb
 - lib/prescient/client.rb
+- lib/prescient/configuration_loader.rb
 - lib/prescient/errors.rb
 - lib/prescient/pgvector.rb
 - lib/prescient/provider/anthropic.rb
+- lib/prescient/provider/deepseek.rb
+- lib/prescient/provider/gemini.rb
 - lib/prescient/provider/huggingface.rb
+- lib/prescient/provider/mistral.rb
 - lib/prescient/provider/ollama.rb
 - lib/prescient/provider/openai.rb
+- lib/prescient/provider/xai.rb
 - lib/prescient/version.rb
+- schema/prescient.configuration.schema.json
 - scripts/setup-ollama-models.sh
 - sig/prescient.rbs
 homepage: https://kanutocd.github.io/prescient