prescient 0.6.0 → 0.8.0

data/sig/prescient.rbs

@@ -33,6 +33,224 @@ module Prescient
   class ProviderError < Error
   end
 
+  class ToolError < Error
+  end
+
+  class ToolConfigurationError < ToolError
+  end
+
+  class ToolConnectionError < ToolError
+  end
+
+  class ToolInvalidResponseError < ToolError
+  end
+
+  module Tool
+    class Base
+      DEFAULT_TIMEOUT: Integer
+      DEFAULT_MAX_RESULTS: Integer
+      MAX_QUERY_LENGTH: Integer
+      DEFAULT_MAX_RESPONSE_BYTES: Integer
+
+      attr_reader options: Hash[Symbol, untyped]
+
+      def initialize: (**untyped) -> void
+      def search: (String, **untyped) -> Hash[Symbol, untyped]
+    end
+
+    class Group < Base
+      attr_reader adapters: Array[Base]
+      def initialize: (adapters: Array[Base]) -> void
+      def search: (String, **untyped) -> Hash[Symbol, untyped]
+    end
+
+    class SearXNG < Base
+      def initialize: (**untyped) -> void
+      def search: (String, ?limit: Integer?, ?timeout: Numeric?) -> Hash[Symbol, untyped]
+    end
+
+    class SearchApi < Base
+      API_URL: String
+      def initialize: (**untyped) -> void
+      def search: (String, ?limit: Integer?, ?timeout: Numeric?) -> Hash[Symbol, untyped]
+    end
+  end
+
+  module DocumentSource
+    DEFAULT_MAX_DOCUMENTS: Integer
+    DEFAULT_MAX_BYTES: Integer
+
+    class Base
+      def initialize: (?max_documents: Integer, ?max_bytes: Integer) -> void
+      def fetch: () -> Array[Hash[String, untyped]]
+    end
+
+    class Memory < Base
+      def initialize: (documents: Array[Hash[String, untyped]] | Hash[String, untyped], ?max_documents: Integer, ?max_bytes: Integer) -> void
+      def fetch: () -> Array[Hash[String, untyped]]
+    end
+
+    class JsonFile < Base
+      def initialize: (path: String, ?max_documents: Integer, ?max_bytes: Integer) -> void
+      def fetch: () -> Array[Hash[String, untyped]]
+    end
+
+    class RedisJson < Base
+      def initialize: (client: untyped, key: String, ?max_documents: Integer, ?max_bytes: Integer) -> void
+      def fetch: () -> Array[Hash[String, untyped]]
+    end
+  end
+
+  module Agent
+    class Error < Prescient::Error
+    end
+    class ConfigurationError < Error
+    end
+    class MalformedActionError < Error
+    end
+    class UnauthorizedToolError < Error
+    end
+    class MaxLoopsExceededError < Error
+    end
+
+    class ErrorSerializer
+      def self.serialize: (Exception) -> Hash[Symbol, untyped]
+    end
+
+    class SchemaValidator
+      def self.validate!: (Hash[untyped, untyped], untyped) -> untyped
+      def initialize: (Hash[untyped, untyped]) -> void
+      def validate!: (untyped) -> untyped
+    end
+
+    class Configuration
+      DEFAULT_MAX_LOOPS: Integer
+      DEFAULT_MAX_CONTEXT_BYTES: Integer
+      DEFAULT_MAX_ACTION_BYTES: Integer
+      DEFAULT_MAX_OBSERVATION_BYTES: Integer
+      DEFAULT_MAX_TASK_BYTES: Integer
+      DEFAULT_MAX_RESPONSE_BYTES: Integer
+
+      attr_reader max_loops: Integer
+      attr_reader max_context_bytes: Integer
+      attr_reader max_action_bytes: Integer
+      attr_reader max_observation_bytes: Integer
+      attr_reader max_task_bytes: Integer
+      attr_reader max_response_bytes: Integer
+
+      attr_reader authorization: Proc?
+      attr_reader telemetry: Proc?
+      def initialize: (?max_loops: Integer, ?max_context_bytes: Integer, ?max_action_bytes: Integer, ?max_observation_bytes: Integer, ?max_task_bytes: Integer, ?max_response_bytes: Integer, ?authorization: Proc?, ?telemetry: Proc?) -> void
+    end
+
+    class AuditLog
+      FIELDS: Array[Symbol]
+      def initialize: (?path: String?, ?io: untyped) -> void
+      def call: (Hash[Symbol, untyped]) -> void
+    end
+
+    class Context
+      attr_reader messages: Array[Hash[Symbol, String]]
+      def initialize: (system_prompt: String, task: String, ?max_bytes: Integer) -> void
+      def append: (role: String, content: String) -> void
+      def to_a: () -> Array[Hash[Symbol, String]]
+    end
+
+    class Parser
+      ACTION_PATTERN: Regexp
+      def self.parse: (String, ?max_bytes: Integer) -> Hash[Symbol, untyped]?
+      def initialize: (?max_bytes: Integer) -> void
+      def parse: (String) -> Hash[Symbol, untyped]?
+    end
+
+    class PromptBuilder
+      def self.build: (system_instruction: String, tools: Array[ToolRegistry::Tool]) -> String
+    end
+
+    class ToolRegistry
+      class Tool
+        attr_reader name: Symbol
+        attr_reader description: String
+        attr_reader schema: Hash[untyped, untyped]
+        attr_reader callable: Proc
+      end
+
+      def initialize: (Hash[Symbol | String, untyped]) -> void
+      def all: () -> Array[Tool]
+      def invoke: (Symbol | String, Hash[String | Symbol, untyped]) -> untyped
+    end
+
+    class Result
+      attr_reader response: String
+      attr_reader provider: Symbol | String | nil
+      attr_reader model: String | nil
+      attr_reader loops_run: Integer
+      attr_reader metadata: Hash[untyped, untyped]
+      def initialize: (response: String, provider: Symbol | String | nil, model: String | nil, loops_run: Integer, ?success: bool, ?metadata: Hash[untyped, untyped]) -> void
+      def success?: () -> bool
+      def to_h: () -> Hash[Symbol, untyped]
+    end
+
+    class Runtime
+      def initialize: (?provider: Symbol?, ?client: untyped, ?tools: Hash[Symbol | String, untyped]?, ?tool_names: Array[Symbol | String], ?configuration: Configuration, ?system_prompt: String, ?provider_options: Hash[Symbol, untyped], ?generation_options: Hash[Symbol, untyped], ?authorization: Proc?, ?telemetry: Proc?, ?enable_fallback: bool, ?request_context: Hash[Symbol, untyped], ?audit_log: AuditLog?) -> void
+      def run: (String) -> Result
+    end
+
+    class CLIAdapter
+      def initialize: (?output: untyped, ?errors: untyped) -> void
+      def run: (task: String, ?client: untyped, ?provider: Symbol?, ?tool_names: Array[Symbol | String], ?max_loops: Integer, ?format: String, ?provider_options: Hash[Symbol, untyped], ?generation_options: Hash[Symbol, untyped], ?telemetry: Proc?) -> Integer
+    end
+  end
+
+  module MCP
+    interface _Authenticator
+      def call: (Hash[String, untyped]) -> untyped
+    end
+
+    class Configuration
+      DEFAULT_MAX_INPUT_BYTES: Integer
+      DEFAULT_TOOLS: Array[String]
+      SUPPORTED_RESOURCES: Array[String]
+      attr_reader name: String
+      attr_reader version: String
+      attr_reader max_input_bytes: Integer
+      attr_reader tools: Array[String]
+      attr_reader resources: Array[String]
+      def initialize: (?name: String, ?version: String, ?max_input_bytes: Integer, ?tools: Array[String], ?resources: Array[String]) -> void
+    end
+
+    class Server
+      def initialize: (?configuration: Configuration, ?client_factory: Proc?, ?authorization: Proc?) -> void
+      def initialize_result: () -> Hash[Symbol, untyped]
+      def dispatch: (Hash[String, untyped], ?context: Hash[Symbol, untyped]) -> Hash[Symbol, untyped]
+      def tools: () -> Array[Hash[String, untyped]]
+      def resources: () -> Array[Hash[String, untyped]]
+      def call_tool: (String | Symbol, ?Hash[String, untyped], ?context: Hash[Symbol, untyped]) -> Hash[Symbol, untyped]
+      def read_resource: (String | Symbol) -> Hash[Symbol, untyped]
+    end
+
+    class Stdio
+      def initialize: (?server: Server, ?input: untyped, ?output: untyped) -> void
+      def run: () -> void
+    end
+
+    class Rack
+      PROTOCOL_VERSION: String
+      SESSION_HEADER: String
+      PROTOCOL_HEADER: String
+      ALLOWED_ORIGINS_DEFAULT: Array[String]
+      def initialize: (authentication: _Authenticator, ?server: Server, ?request_context: Proc?, ?max_body_bytes: Integer, ?allowed_origins: Array[String], ?session_ttl: Numeric) -> void
+      def call: (Hash[String, untyped]) -> Array[untyped]
+    end
+
+    module Authentication
+      class BearerToken
+        def initialize: (token: String, ?principal: untyped) -> void
+        def call: (Hash[String, untyped]) -> untyped
+      end
+    end
+  end
+
   class Base
     attr_reader options: Hash[Symbol, untyped]
     attr_reader provider_name: String
@@ -257,11 +475,15 @@ module Prescient
     attr_accessor fallback_providers: Array[Symbol]
     attr_reader sensitive_keys: Array[Symbol]
     attr_reader providers: Hash[Symbol, Hash[Symbol, untyped]]
+    attr_reader tools: Hash[Symbol, Hash[Symbol, untyped]]
 
     def initialize: () -> void
     def sensitive_keys=: (Array[Symbol | String]) -> Array[Symbol]
     def add_provider: (Symbol | String, Class, **untyped) -> void
     def provider: (Symbol | String) -> untyped
+    def add_tool: (Symbol | String, Class, **untyped) -> void
+    def add_tool_group: (Symbol | String, Array[Hash[Symbol, untyped]]) -> void
+    def tool: (Symbol | String) -> untyped
     def available_providers: () -> Array[Symbol]
   end
 
@@ -304,7 +526,7 @@ module Prescient
     API_VERSION: String
     ROUTES: Hash[Array[String], Symbol]
 
-    def initialize: (?authentication: untyped, ?max_body_bytes: Integer) -> void
+      def initialize: (?authentication: untyped, ?authorization: untyped, ?request_context: untyped, ?telemetry: Proc?, ?max_body_bytes: Integer) -> void
     def call: (Hash[String, untyped]) -> [Integer, Hash[String, String], Array[String]]
   end
 
@@ -313,6 +535,8 @@ module Prescient
     TOP_LEVEL_KEYS: Array[String]
     PROVIDER_TYPES: Hash[String, untyped]
     COMMON_PROVIDER_KEYS: Array[String]
+    TOOL_TYPES: Hash[String, untyped]
+    COMMON_TOOL_KEYS: Array[String]
     PROMPT_TEMPLATE_KEYS: Array[String]
     ATTR_KEYS: Array[String]
 
@@ -335,11 +559,14 @@ module Prescient
   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]
+  def self.search_and_generate: (String, ?tool: Symbol | String, ?provider: Symbol, ?limit: Integer?, ?enable_fallback: bool, ?provider_options: Hash[Symbol, untyped], **untyped) -> Hash[Symbol, untyped]
   def self.health_check: (?provider: Symbol) -> Hash[Symbol, untyped]
+  def self.tool: (Symbol | String) -> untyped
 
   private
 
   def self.configure_default_providers: (Configuration, untyped) -> void
+  def self.configure_default_tools: (Configuration, untyped) -> void
   def self.configure_ollama: (Configuration, untyped) -> void
   def self.configure_openai: (Configuration, untyped) -> void
   def self.configure_anthropic: (Configuration, untyped) -> void

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: prescient
 version: !ruby/object:Gem::Version
-  version: 0.6.0
+  version: 0.8.0
 platform: ruby
 authors:
 - Ken C. Demanawa
@@ -23,13 +23,15 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: 0.24.0
-description: "Prescient provides a consistent interface for AI providers including
-  OpenAI, Anthropic, \nOllama, and Hugging Face, with provider selection, retries,
-  health checks, and automatic fallback.\n"
+description: |
+  Prescient provides a consistent Ruby API, CLI, or REST API for AI providers including
+  Ollama, OpenAI, Anthropic, Hugging Face, Google Gemini, Mistral, DeepSeek, and xAI,
+  with provider selection, retries, health checks, and fallback across configured providers.
 email:
 - kenneth.c.demanawa@gmail.com
 executables:
 - prescient
+- prescient-mcp
 extensions: []
 extra_rdoc_files: []
 files:
@@ -60,14 +62,36 @@ files:
 - examples/custom_prompts.rb
 - examples/rest_api.ru
 - examples/vector_search.rb
+- examples/web_search.rb
 - exe/prescient
+- exe/prescient-mcp
 - lib/prescient.rb
+- lib/prescient/agent.rb
+- lib/prescient/agent/audit_log.rb
+- lib/prescient/agent/cli_adapter.rb
+- lib/prescient/agent/configuration.rb
+- lib/prescient/agent/context.rb
+- lib/prescient/agent/error_serializer.rb
+- lib/prescient/agent/errors.rb
+- lib/prescient/agent/parser.rb
+- lib/prescient/agent/prompt_builder.rb
+- lib/prescient/agent/result.rb
+- lib/prescient/agent/runtime.rb
+- lib/prescient/agent/schema_validator.rb
+- lib/prescient/agent/tool_registry.rb
 - lib/prescient/api.rb
 - lib/prescient/base.rb
 - lib/prescient/cli.rb
 - lib/prescient/client.rb
 - lib/prescient/configuration_loader.rb
+- lib/prescient/document_source.rb
 - lib/prescient/errors.rb
+- lib/prescient/mcp.rb
+- lib/prescient/mcp/authentication.rb
+- lib/prescient/mcp/configuration.rb
+- lib/prescient/mcp/rack.rb
+- lib/prescient/mcp/server.rb
+- lib/prescient/mcp/stdio.rb
 - lib/prescient/pgvector.rb
 - lib/prescient/provider/anthropic.rb
 - lib/prescient/provider/deepseek.rb
@@ -77,9 +101,13 @@ files:
 - lib/prescient/provider/ollama.rb
 - lib/prescient/provider/openai.rb
 - lib/prescient/provider/xai.rb
+- lib/prescient/tool.rb
+- lib/prescient/tool/search_api.rb
+- lib/prescient/tool/searxng.rb
 - lib/prescient/version.rb
 - schema/prescient.configuration.schema.json
 - scripts/setup-ollama-models.sh
+- searxng/settings.yml
 - sig/prescient.rbs
 homepage: https://kanutocd.github.io/prescient
 licenses:
@@ -106,5 +134,5 @@ required_rubygems_version: !ruby/object:Gem::Requirement
 requirements: []
 rubygems_version: 4.0.16
 specification_version: 4
-summary: A boring AI provider abstraction for Ruby
+summary: A boring AI provider gateway for Ruby
 test_files: []