clag 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 29a6ab190bdd6accfd67782655a99b3f255c0d5b217a18fa41ddd2bf1f837dea
-  data.tar.gz: 6ad8f31dcebbcc91a8e17d1cb4a90de81ce93f91799de6a7e7e3abd721f15017
+  metadata.gz: 590439d5a481273f60f73e23d193490499a0bc6f9c400fa6239f00e790f31191
+  data.tar.gz: 2b35f0ef8b99e0185761275a405aa44363380df88a0f53449a3e3e1208159659
 SHA512:
-  metadata.gz: b7751ed524c187b0925610419caaedce08605caaea43df5066def060dfd8fcb7ee112afa0c8185ef684103d9924f6b06d24475fe81d2ee552ebe9b698fad9336
-  data.tar.gz: d6ba64383b5500c849179203b4aac5f4d88ecd048319b0a9ce2a87946cb8a8c7e558316b5763d660a329eb1a201fd6052278429fbb5a657f7cce0782df5295a2
+  metadata.gz: 87982f697b62eb353350fea66026841e8b0815bff79fea7407b7d3159ecc85aaefad4fab52a0c0f40dbd5389f5b83fa47b86dd6951de7fbd8ba33de58beae4a4
+  data.tar.gz: 637a5801e9e52443e7bd694fedbef83f1baa9fd624ad263c9abe00f2874760a4ec3bccb5205bf14e33e1d50df323dc921cc8a52df3541f22849fc7640ece90cf

data/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 Sublayer
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

data/README.md

@@ -8,16 +8,28 @@ command with the help of an LLM!
 
 ## Installation
 
-* Get an API key from OpenAI for gpt4-turbo: https://platform.openai.com/
-
-* Set your API key as OPENAI\_API\_KEY in your environment
-
 * Install the gem
 `gem install clag`
 
 * Generate commands
 `clag g "create a new ruby on rails project using postgres and tailwindcss"`
 
+### Using OpenAI's GPT-4
+
+* Get an API key from OpenAI for gpt4-turbo: https://platform.openai.com/
+
+* Set your API key as OPENAI\_API\_KEY in your environment
+
+
+### Using Google's Gemini 1.0
+
+* Get an API key from Google's AI Studio at https://ai.google.dev/
+
+* Set your API key as GEMINI\_API\_KEY in your environment
+
+* Select Gemini as your preferred LLM by setting CLAG\_LLM=gemini in your
+  environment
+
 ## Usage
 
 Currently support one command: "g".
@@ -28,3 +40,9 @@ Currently support one command: "g".
 
 Bug reports and pull requests are welcome on Github at
 https://github.com/sublayerapp/clag
+
+## Community
+
+Like what you see, or looking for more people working on the future of
+programming with LLMs? Come join us in the [Promptable Architecture
+Discord](https://discord.gg/sjTJszPwXt)

data/clag.gemspec

@@ -20,6 +20,7 @@ Gem::Specification.new do |spec|
   spec.add_dependency 'cli-kit', '~> 5'
   spec.add_dependency 'cli-ui', '~> 2.2.3'
   spec.add_dependency 'ruby-openai', '~> 6'
+  spec.add_dependency 'httparty', '~> 0.21'
   spec.add_dependency 'clipboard', '~> 1.3'
   spec.add_dependency 'activesupport'
 

data/lib/clag/version.rb

@@ -1,3 +1,3 @@
 module Clag
-  VERSION = '0.0.1'
+  VERSION = '0.0.2'
 end

data/vendor/gems/sublayer/lib/sublayer/capabilities/llm_assistance.rb

@@ -1,4 +1,5 @@
 require "openai"
+require "httparty"
 
 module Sublayer
   module Capabilities
@@ -15,6 +16,53 @@ module Sublayer
       end
 
       def llm_generate
+        if ENV["CLAG_LLM"] == "gemini"
+          generate_with_gemini
+        else
+          generate_with_openai
+        end
+      end
+
+      private
+      def generate_with_gemini
+        gemini_prompt = adapt_prompt_for_gemini
+
+        response = HTTParty.post("https://generativelanguage.googleapis.com/v1beta/models/gemini-pro:generateContent?key=#{ENV['GEMINI_API_KEY']}", body: gemini_prompt.to_json, headers: { 'Content-Type' => 'application/json' })
+
+        function_output = extract_function_output_from_gemini_response(response)
+
+        return function_output
+      end
+
+      def adapt_prompt_for_gemini
+        return({ tools: { function_declarations: [ self.class::OUTPUT_FUNCTION.to_hash ] }, contents:  { role: "user", parts: { text: prompt } }  })
+      end
+
+      def extract_function_output_from_gemini_response(response)
+        candidates = response.dig('candidates')
+        if candidates && candidates.size > 0
+          content = candidates[0].dig('content')
+          if content && content['parts'] && content['parts'].size > 0
+            part = content['parts'][0]
+
+            # Check if the part contains a function call
+            if part.key?('functionCall')
+              function_name = part['functionCall']['name']
+              args = part['functionCall']['args']
+
+              # Assuming the agent expects a single string parameter:
+              if args && args.key?(self.class::OUTPUT_FUNCTION.name)
+                return args[self.class::OUTPUT_FUNCTION.name]
+              end
+            else
+              # If it's not a function call, check for a direct string response
+              return part['text'] if part.key?('text')
+            end
+          end
+        end
+      end
+
+      def generate_with_openai
         client = OpenAI::Client.new(access_token: ENV["OPENAI_API_KEY"])
 
         response = client.chat(

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: clag
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - Scott Werner
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-01-26 00:00:00.000000000 Z
+date: 2024-02-21 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: cli-kit
@@ -52,6 +52,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '6'
+- !ruby/object:Gem::Dependency
+  name: httparty
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.21'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.21'
 - !ruby/object:Gem::Dependency
   name: clipboard
   requirement: !ruby/object:Gem::Requirement
@@ -105,6 +119,7 @@ extra_rdoc_files: []
 files:
 - ".gitignore"
 - Gemfile
+- LICENSE
 - README.md
 - bin/testunit
 - clag.gemspec