llmed 0.1.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: f3c14772724d875ead130cb19f08a232b34e197d18a27f062695bbb0bf14e091
+  data.tar.gz: b6e061736119a7c1550d45a517f1dec49206be09a2b0d63ce75d49fb5ccb0787
+SHA512:
+  metadata.gz: 7e7e7ae58ce1b736bbc38973b36263924d434da42a11bf5230e3565c2fd2d5f44d4f150e326910c2b6cedc5c2872714ad95c4f2fc4e3a02a8f79c66d1ecd7124
+  data.tar.gz: 72afee21c7172ce2ff848bdbdf920aea4f5fcd76a339b2039ca588f9be2e88ec089a1e07982ef6288724a1a93c61a56b9468daac9cc240d51b9129c459d8f0b0

data/exe/llmed

@@ -0,0 +1,50 @@
+#!/bin/env ruby
+# Copyright 2025 Jovany Leandro G.C <bit4bit@riseup.net>
+# frozen_string_literal: true
+require 'optparse'
+require 'llmed'
+
+logger = Logger.new(STDOUT)
+output_dir = './llmed-out'
+release_dir = output_dir
+template = <<-TMP
+set_llm provider: :openai, api_key: ENV['OPENAI_API_KEY'], model: 'gpt-4o'
+
+application "hi world", language: '<HERE LANGUAGE>', output_file: "<HERE NAME>.rb" do
+  context "main" do
+    <<-LLM
+        Application do print 'hi world!'.
+    LLM
+  end
+end
+TMP
+
+OptionParser.new do |parser|
+  parser.banner = "Usage: llmed [options]"
+  parser.on_tail("-h", "--help", "Show this message") do
+    puts parser
+    exit
+  end
+
+  parser.on('-t', '--template PATH', String, 'Create template') do |path|
+    File.write path, template
+    exit
+  end
+
+  parser.on('--output-dir DIR', String) do |path|
+    output_dir = path
+  end
+
+  parser.on('--release-dir DIR', String) do |path|
+    release_dir = path
+  end
+end
+
+source_path = ARGF.read
+if ARGF.respond_to?(:path)
+  release_dir = File.dirname(ARGF.path)
+end
+
+llmed = LLMed.new(logger: logger)
+llmed.eval_source source_path
+llmed.compile(output_dir: output_dir, release_dir: release_dir)

data/lib/llmed.rb

@@ -0,0 +1,221 @@
+# Copyright 2025 Jovany Leandro G.C <bit4bit@riseup.net>
+# frozen_string_literal: true
+
+require 'pp'
+require 'langchain'
+require 'pathname'
+require 'fileutils'
+require 'forwardable'
+
+Langchain.logger.level = Logger::ERROR
+
+class LLMed
+  extend Forwardable
+
+  class Context
+    attr_reader :name
+
+    def initialize(name:, options: {})
+      @name = name
+      @skip = options[:skip] || false
+    end
+
+    def skip?
+      @skip
+    end
+
+    def message
+      "# #{@name}\n\n#{@message}"
+    end
+
+    def llm(message)
+      @message = message
+    end
+
+    def message?
+      not (@message.nil? || @message.empty?)
+    end
+
+
+    def from_file(path)
+      File.read(path)
+    end
+
+    def from_source_code(path)
+      code = File.read(path)
+      "Dado el codigo fuente: #{code}\n\n\n"
+    end
+  end
+
+  class Configuration
+    def initialize
+      @prompt = Langchain::Prompt::PromptTemplate.new(template: "
+Eres desarrollador de software y solo conoces del lenguage de programacion {language}.
+La respuesta no debe contener texto adicional al codigo fuente generado.
+Todo el codigo fuente se genera en un unico archivo y debes asegurarte de que se ejecute correctamente desde el primer intento.
+Siempre adicionas el comentario de codigo correctamente escapado LLMED-COMPILED.
+
+Debes solo modificar el siguiente codigo fuente: {source_code}.
+
+", input_variables: ["language", "source_code"])
+    end
+
+    def prompt(language:, source_code:)
+      @prompt.format(language: language, source_code: source_code)
+    end
+
+    def set_prompt(prompt)
+      @prompt = Langchain::Prompt::PromptTemplate.new(template: prompt, input_variables: ["language", "source_code"])
+    end
+
+    def set_language(language)
+      @language = language
+    end
+
+    def set_llm(provider:, api_key:, model:)
+      @provider = provider
+      @provider_api_key = api_key
+      @provider_model = model
+    end
+
+    def language(main)
+      lang = main || @language
+      raise "Please assign a language to the application or general with the function set_languag" if lang.nil?
+      lang
+    end
+
+    def llm()
+      case @provider
+      when :openai
+        Langchain::LLM::OpenAI.new(
+          api_key: @provider_api_key,
+          default_options: { temperature: 0.7, chat_model: @provider_model}
+        )
+      when nil
+        raise "Please set the provider with `set_llm(provider, api_key, model)`"
+      else
+        raise "not implemented provider #{@provider}"
+      end
+    end
+  end
+
+  class Application
+    attr_reader :contexts, :name, :language
+
+    def initialize(name:, language:, output_file:, block:, logger:, release:)
+      raise "required language" if language.nil?
+
+      @name = name
+      @output_file = output_file
+      @language = language
+      @block = block
+      @contexts = []
+      @logger = logger
+      @release = release
+    end
+
+    def context(name, **opts, &block)
+      ctx = Context.new(name: name, options: opts)
+      output = ctx.instance_eval(&block)
+      unless ctx.message?
+        ctx.llm(output)
+      end
+
+      @contexts << ctx
+    end
+
+    def evaluate
+      self.instance_eval(&@block)
+    end
+
+    def source_code(output_dir, release_dir)
+      return unless @release
+      release_source_code = Pathname.new(release_dir) + "#{@output_file}.r#{@release}#{@language}"
+      output_file = Pathname.new(output_dir) + @output_file
+      if @release and not File.exist?(release_source_code)
+        FileUtils.cp(output_file, release_source_code)
+        @logger.info("APPLICATION #{@name} RELEASE FILE #{release_source_code}")
+      end
+
+      return File.read(release_source_code)
+    end
+
+    def output_file(output_dir, mode = 'w')
+      if @output_file.respond_to? :write
+        yield @output_file
+      else
+        path = Pathname.new(output_dir) + @output_file
+        FileUtils.mkdir_p(File.dirname(path))
+
+        @logger.info("APPLICATION #{@name} OUTPUT FILE #{path}")
+
+        File.open(path, mode) do |file|
+          yield file
+        end
+      end
+    end
+  end
+
+  def initialize(logger:)
+    @logger = logger
+    @applications = []
+    @configuration = Configuration.new()
+  end
+
+  def eval_source(code)
+    self.instance_eval(code)
+  end
+
+  # changes default language
+  def_delegator :@configuration, :set_language, :set_language
+  # changes default llm
+  def_delegator :@configuration, :set_llm, :set_llm
+  # changes default prompt
+  def_delegator :@configuration, :set_prompt, :set_prompt
+
+  def application(name, language: nil, release: nil, output_file:, &block)
+    @app = Application.new(name: name, language: @configuration.language(language), output_file: output_file, block: block, logger: @logger, release: release)
+    @applications << @app
+  end
+
+  def compile(output_dir:, release_dir: nil)
+    release_dir = output_dir unless release_dir
+    @applications.each do |app|
+      llm = @configuration.llm()
+      system_content = @configuration.prompt(language: app.language, source_code: app.source_code(output_dir, release_dir))
+      messages = [
+        {role: "system", content: system_content},
+      ]
+      app.evaluate
+      app.contexts.each do |ctx|
+        next if ctx.skip?
+        messages << {role: "user", content: ctx.message}
+      end
+
+      llm_response = llm.chat(messages: messages)
+      response = llm_response.chat_completion
+      @logger.info("APPLICATION #{app.name} TOTAL TOKENS #{llm_response.total_tokens}")
+      write_output(app, output_dir, source_code(response))
+    end
+  end
+
+  private
+  def source_code(content)
+    # TODO: by provider?
+    content.gsub('```', '').sub(/^(ruby|python(\d*)|elixir|c(pp)?|perl|bash)/, '')
+  end
+
+  def write_output(app, output_dir, output)
+    app.output_file(output_dir) do |file|
+      file.write(output)
+    end
+  end
+
+  def edit_same_source_code(app, output_dir, messages)
+    content = ""
+    app.output_file(output_dir, 'r') do |file|
+      content = "Codigo fuente a modificar: #{file.read()}"
+    end
+    messages << {role: "user", content: content}
+  end
+end

metadata

@@ -0,0 +1,104 @@
+--- !ruby/object:Gem::Specification
+name: llmed
+version: !ruby/object:Gem::Version
+  version: 0.1.1
+platform: ruby
+authors:
+- Jovany Leandro G.C
+autorequire:
+bindir: exe
+cert_chain: []
+date: 2025-06-04 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: langchainrb
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.19.5
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.19.5
+- !ruby/object:Gem::Dependency
+  name: ruby-openai
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '8.1'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '8.1'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.13'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.13'
+- !ruby/object:Gem::Dependency
+  name: rubocop
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.75'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.75'
+description: LLM Execution Development
+email: bit4bit@riseup.net
+executables:
+- llmed
+extensions: []
+extra_rdoc_files: []
+files:
+- exe/llmed
+- lib/llmed.rb
+homepage: https://github.com/bit4bit/llm-labs/tree/main/llmed
+licenses:
+- GPL-3.0
+metadata:
+  source_code_uri: https://github.com/bit4bit/llm-labs
+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.15
+signing_key:
+specification_version: 4
+summary: Use this 'compiler' to build software using LLMs in a controlled way. In
+  classical terms, the LLM is the compiler, the context description is the programming
+  language, and the generated output is the binary.
+test_files: []