Class: Llm::Functions::OverwriteFile

Inherits:
Base
  • Object
show all
Defined in:
lib/llm/functions/overwrite_file.rb

Instance Method Summary collapse

Methods inherited from Base

#stop_llm_call?

Instance Method Details

#definition ⇒ Object 



8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
 
# File 'lib/llm/functions/overwrite_file.rb', line 8

def definition
  return @definition unless @definition.nil?

  @definition = {
    name: self.function_name,
    description: I18n.t("ghostest.functions.#{self.function_name}.description"),
    parameters: {
      type: :object,
      properties: {
        filepath: {
          type: :string,
          description: I18n.t("ghostest.functions.#{self.function_name}.parameters.filepath"),
        },
        new_text: {
          type: :string,
          description: I18n.t("ghostest.functions.#{self.function_name}.parameters.new_text"),
        },
      },
      required: [:filepath, :new_text],
    },
  }
  @definition
end

#execute_and_generate_message(args) ⇒ Object 



32
33
34
35
36
37
38
39
 
# File 'lib/llm/functions/overwrite_file.rb', line 32

def execute_and_generate_message(args)
  unless File.exist?(args[:filepath])
    raise Ghostest::Error.new("File not found: #{args[:filepath]}")
  end
  File.write(args[:filepath], args[:new_text])

  { result: :success }
end

#function_name ⇒ Object 



4
5
6
 
# File 'lib/llm/functions/overwrite_file.rb', line 4

def function_name
  :overwrite_file
end