ollama-find 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 3852b0b7aa8d784ce62440069312aaeb3479013be0bf95bd6b681f6eccec2f64
+  data.tar.gz: 3dd15be963890f5904142f6d1b050c167a43cd914ea124803e06819f78100b4f
+SHA512:
+  metadata.gz: faf0b9412b0105e14ae79de3d4df119e7451a68443d382a7d36544025fa0b2a9bb40224f6026fe6890e1fe0b94ad1f70a19db0a0760c24ca13c2642cd3061db6
+  data.tar.gz: 33d9079e414c0f9cf18f6d6563fff26e863f6b223cd08a330ffa97babb05dd32be7da08036fb301e5e74da63685b65744f7a07e14362647e393f944dbbab1e20

data/CODE_OF_CONDUCT.md

@@ -0,0 +1,13 @@
+# Contributor Code of Conduct
+
+As contributors and maintainers of this project, we pledge to respect all people who contribute through reporting issues, posting feature requests, updating documentation, submitting pull requests or patches, and other activities.
+
+We are committed to making participation in this project a harassment-free experience for everyone, regardless of level of experience, gender, gender identity and expression, sexual orientation, disability, personal appearance, body size, race, age, or religion.
+
+Examples of unacceptable behavior by participants include the use of sexual language or imagery, derogatory comments or personal attacks, trolling, public or private harassment, insults, or other unprofessional conduct.
+
+Project maintainers have the right and responsibility to remove, edit, or reject comments, commits, code, wiki edits, issues, and other contributions that are not aligned to this Code of Conduct. Project maintainers who do not follow the Code of Conduct may be removed from the project team.
+
+Instances of abusive, harassing, or otherwise unacceptable behavior may be reported by opening an issue or contacting one or more of the project maintainers.
+
+This Code of Conduct is adapted from the [Contributor Covenant](http:contributor-covenant.org), version 1.0.0, available at [http://contributor-covenant.org/version/1/0/0/](http://contributor-covenant.org/version/1/0/0/)

data/README.md

@@ -0,0 +1,23 @@
+# Ollama::Find
+
+This is a CLI tool that allows you to quickly generate a path to a gguf file that's been pulled via Ollama.
+
+
+## Installation
+
+```bash
+gem install ollama-find
+```
+
+
+## Usage
+
+This guide assumes you've already installed [Ollama](https://ollama.com/download) and it's available on your command line.
+
+```bash
+# Pull a model to test with
+ollama pull codegpt/replit-code-v1_5-3b
+
+# Get the path to that model
+ollama-find codegpt/replit-code-v1_5-3b
+```

data/Rakefile

@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+require "bundler/gem_tasks"
+require "rspec/core/rake_task"
+
+RSpec::Core::RakeTask.new(:spec) do |t|
+  t.verbose = false
+end
+
+RSpec::Core::RakeTask.new(:integration) do |t|
+  t.verbose = false
+  t.pattern = "spec/**/*_int.rb"
+end
+
+# this is for running tests that you've marked current... eg: it 'should work', :current => true do
+# I waffle between this work flow and the one defined in spec_helper.rb which uses focus: true
+# which overrides the entire test suite... not better
+RSpec::Core::RakeTask.new(:current) do |t|
+  t.verbose = false
+  t.pattern = 'spec/**/*_{spec,int}.rb'
+  t.rspec_opts = ['--tag current']
+end
+
+# alias for current
+RSpec::Core::RakeTask.new(:c) do |t|
+  t.verbose = false
+  t.pattern = 'spec/**/*_{spec,int}.rb'
+  t.rspec_opts = ['--tag current']
+end
+
+task default: :spec

data/changelog

@@ -0,0 +1,7 @@
+** 0.0.x **
+  TODO:  List off what features you'll be including at some point in the
+future.  
+
+** 0.0.1 **
+  TODO:  List off what features you'll be including in the first release
+

data/exe/ollama-find

@@ -0,0 +1,25 @@
+#!/usr/bin/env ruby
+
+require "ollama-find"
+
+arg0 = ARGV.count > 0 ? ARGV[0].chomp.strip : ''
+arg1 = ARGV.count > 1 ? ARGV[1].chomp.strip : nil
+
+if ARGV.count == 0 || arg0 == "help" || arg0 == "--help" || arg0 == "-h"
+  puts Ollama::Find.help
+  return
+end
+
+model_name = arg0
+model_tag = arg1
+
+# Check if model_tag is specified in first argument via ':' symbol
+if arg0.include?(":")
+  model_name = arg0.split(":")[0]
+  model_tag = arg0.split(":")[1]
+end
+
+# if no model_tag, default to latest
+model_tag = 'latest' if model_tag.nil?
+
+puts Ollama::Find.main(model_name, model_tag)

data/gitignore

@@ -0,0 +1,8 @@
+/.bundle/
+/.yardoc
+/_yardoc/
+/coverage/
+/doc/
+/pkg/
+/spec/reports/
+/tmp/

data/lib/ollama-find/lookup_gguf_path.rb

@@ -0,0 +1,98 @@
+require 'yaml'
+require 'json'
+
+
+module Ollama
+  module Find
+    CLEAN_MODEL_DIR = "~/.ollama/models"
+    MODEL_DIR = File.expand_path(CLEAN_MODEL_DIR)
+
+    # Retrieves the file path to the GGUF model based on a given model name and optional tag.
+    #
+    # @param model_uri [String] Just the name of the model (e.g. "mymodel") or a full
+    #                           uri including registry and model (e.g. "registry.com/user/mymodel").
+    # @param model_tag [String] (Optional) The model version/tag. Defaults to "latest".
+    # @return [String, nil] The absolute path to the GGUF model file, or `nil` if the model digest
+    #                       is not found.
+    #
+    # @raise [RuntimeError] If the manifest file does not exist.
+    def self.lookup_gguf_path(model_uri, model_tag = nil)
+      model_tag = "latest" if model_tag.nil?
+      registry_path, model_name = split_registry_from_model_name(model_uri)
+      path_to_manifest = File.join(MODEL_DIR, "manifests", registry_path, model_name, model_tag)
+
+      if file_missing?(path_to_manifest)
+        msg = "Error: Manifest for #{model_name} could not be found.  Checked #{clean_path(path_to_manifest)}"
+        if it_looks_like_you_needed_to_supply_a_tag_name?(path_to_manifest)
+          tagged_file = get_example_tagname(path_to_manifest)
+          msg += ".\n\nIf you meant to specify a version, try:\n  $  #{command_name} #{model_uri} #{tagged_file}"
+        end
+        return msg
+      end
+
+      begin
+        manifest = JSON.parse(read_manifest(path_to_manifest))
+      rescue JSON::ParserError
+        return "Error: Unable to parse manifest at #{path_to_manifest}"
+      end
+
+      begin
+        digest = extract_model_digest(manifest)
+      rescue
+        return "Error: Unable to extract digest from manifest at #{path_to_manifest} for model_name #{model_name}"
+      end
+
+      File.join(CLEAN_MODEL_DIR, "blobs", digest)
+    end
+
+    def self.command_name
+      "#{File.basename($0)} find"
+    end
+
+    def self.get_example_tagname(path)
+      dirpath = File.dirname(path)
+      File.basename(Dir.glob("#{dirpath}/*").first)
+    end
+
+    def self.it_looks_like_you_needed_to_supply_a_tag_name?(path)
+      Dir.exist?(File.dirname(path))
+    end
+
+    def self.clean_path(absolute_path)
+      absolute_path.sub(MODEL_DIR, CLEAN_MODEL_DIR)
+    end
+
+    def self.extract_model_digest(manifest)
+      manifest["layers"].select do |obj|
+        obj["mediaType"] == "application/vnd.ollama.image.model"
+      end.first["digest"].sub(':', '-')
+    end
+
+    def self.split_registry_from_model_name(model_name)
+      if model_name.split("/").first.include?(".")
+        return get_private_registry_model_name_and_registry(model_name)
+      end
+      if model_name.split("/").count > 1
+        subcatalog = model_name.split("/").first
+        return ["registry.ollama.ai/#{subcatalog}", model_name.sub("#{subcatalog}/", "")]
+      end
+      ["registry.ollama.ai/library", model_name]
+    end
+
+    def self.get_private_registry_model_name_and_registry(model_name)
+      folder_array = model_name.split("/")
+      model_name = folder_array.pop
+      registry_path = folder_array.join("/")
+      [registry_path, model_name]
+    end
+
+    def self.read_manifest(path_to_manifest)
+      File.read(File.expand_path(path_to_manifest))
+    end
+
+    def self.file_missing?(path)
+      !File.exist?(path)
+    end
+
+  end
+end

data/lib/ollama-find/version.rb

@@ -0,0 +1,5 @@
+module Ollama
+  module Find
+    VERSION = "0.1.0"
+  end
+end

data/lib/ollama-find.rb

@@ -0,0 +1,25 @@
+require "ollama-find/version"
+require "ollama-find/lookup_gguf_path"
+
+
+module Ollama
+  module Find
+
+    def self.main(model_uri, model_tag = nil)
+      lookup_gguf_path(model_uri, model_tag)
+    end
+
+    def self.help
+      msg = <<-EOF.strip
+ollama::find
+  A CLI tool that allows you to quickly generate a path to a gguf file that's
+  been pulled via Ollama.
+
+Usage:
+  $  ollama-find llama3
+  ~/.ollama/models/blobs/sha256-6a0746a1ec1aef3e7ec53868f220ff6e389f6f8ef87a01d77c96807de94ca2aa
+  EOF
+    end
+
+  end
+end

data/rspec

@@ -0,0 +1,2 @@
+--format documentation
+--color

metadata

@@ -0,0 +1,110 @@
+--- !ruby/object:Gem::Specification
+name: ollama-find
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- TheNotary
+bindir: exe
+cert_chain: []
+date: 2025-03-30 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.6'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.6'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '13.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '13.0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: pry
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+email:
+- no@email.plz
+executables:
+- ollama-find
+extensions: []
+extra_rdoc_files: []
+files:
+- CODE_OF_CONDUCT.md
+- README.md
+- Rakefile
+- changelog
+- exe/ollama-find
+- gitignore
+- lib/ollama-find.rb
+- lib/ollama-find/lookup_gguf_path.rb
+- lib/ollama-find/version.rb
+- rspec
+homepage: https://github.com/TheNotary/ollama-find
+licenses:
+- MIT
+metadata:
+  homepage_uri: https://github.com/TheNotary/ollama-find
+  source_code_uri: https://github.com/TheNotary/ollama-find
+  changelog_uri: https://github.com/TheNotary/ollama-find
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 3.0.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.6.2
+specification_version: 4
+summary: A CLI tool that helps you quickly generate the path to a gguf file cached
+  by Ollama
+test_files: []