o_collect 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 043f3caa318ad8655c31a45a2056ed575debdb7b9b9985e330096f7b299f4fa9
+  data.tar.gz: 3d1896f697d239180030577e44e8c35bd010311af825cac7d365af44c584cf7e
+SHA512:
+  metadata.gz: a53fe96f695155d92b32e624b68f7ea8473de85c66e414fdaaf73266d797124af44e1422bfa332979d76ca15e95b14ff0dd0fc387ddd123d9b07c0167ef460c7
+  data.tar.gz: a45b96c49b3b9d65913ec0c7deaba1b27b82e3ea1d21478f7d11187f36b035920764b9f692235fcfc1ecd0b039ad029679ddbd6e04123ed4139defa75669cc38

data/bin/o_collect

@@ -0,0 +1,4 @@
+#!/usr/bin/env ruby
+require "collector/o_collect"
+
+Collector::OCollect.run 

data/lib/collector/o_collect.rb

@@ -0,0 +1,80 @@
+require 'yaml'
+require 'time'
+require 'groq'
+include Groq::Helpers
+
+module Collector
+  class OCollect
+    DEFAULT_CONFIG = {
+      "version"    => "0.1",
+      "database"   => "local.db",
+      "target_url" => "https://localhost:8080/collect/test",
+      "prompt"     => "name a random country",
+      "interval"   => 1000
+    }
+
+    CONFIG_FILENAME = ".o_collect"
+
+    API_KEY = "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"
+    MODEL_ID = "mixtral-8x7b-32768"
+    ROLE     = "data scientist"
+
+    def self.run
+      new.run
+    end
+
+    def run
+      ensure_config_file_exists
+      config = load_config
+
+      # Update the config with the current datetime under key "last_run_at"
+      config["last_run_at"] = Time.now.utc.iso8601
+      File.write(CONFIG_FILENAME, config.to_yaml)
+      puts "Configuration updated with last_run_at: #{config['last_run_at']}"
+
+      # Set up Groq Client
+      Groq.configure do |config_obj|
+        config_obj.api_key  = API_KEY
+        config_obj.model_id = MODEL_ID
+      end
+
+      client = Groq::Client.new
+
+      # Create a fiber that repeatedly calls the Groq API
+      fiber = Fiber.new do
+        loop do
+          response = client.chat([
+            System(ROLE),
+            U(config["prompt"])
+          ])
+          puts response['content']
+
+          # Pause for the configured interval (convert ms to seconds)
+          sleep(config["interval"].to_f / 1000)
+
+          Fiber.yield
+        end
+      end
+
+      # Continuously resume the fiber while it is alive
+      while fiber.alive?
+        fiber.resume
+      end
+    end
+
+    private
+
+    def ensure_config_file_exists
+      unless File.exist?(CONFIG_FILENAME)
+        File.open(CONFIG_FILENAME, "w") do |file|
+          file.write(DEFAULT_CONFIG.to_yaml)
+        end
+        puts "Created default configuration file #{CONFIG_FILENAME}"
+      end
+    end
+
+    def load_config
+      YAML.load_file(CONFIG_FILENAME)
+    end
+  end
+end 

data/lib/collector/version.rb

@@ -0,0 +1,3 @@
+module Collector
+  VERSION = "0.1.0"
+end 

metadata

@@ -0,0 +1,62 @@
+--- !ruby/object:Gem::Specification
+name: o_collect
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Han
+autorequire:
+bindir: bin
+cert_chain: []
+date: 2025-02-02 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 0.3.2
+  name: groq
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 0.3.2
+description: Provides the o_collect command which periodically calls the Groq API
+  using a configured prompt.
+email:
+- han@odore.ai
+executables:
+- o_collect
+extensions: []
+extra_rdoc_files: []
+files:
+- bin/o_collect
+- lib/collector/o_collect.rb
+- lib/collector/version.rb
+homepage: https://www.odore.ai
+licenses:
+- Nonstandard
+metadata: {}
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.3.26
+signing_key:
+specification_version: 4
+summary: Command-line tool for periodic Groq API calls
+test_files: []