aboutsummaryrefslogtreecommitdiffstats
path: root/run/tool.rb
diff options
context:
space:
mode:
Diffstat (limited to 'run/tool.rb')
-rwxr-xr-xrun/tool.rb17
1 files changed, 17 insertions, 0 deletions
diff --git a/run/tool.rb b/run/tool.rb
index cf51f3f..2b47788 100755
--- a/run/tool.rb
+++ b/run/tool.rb
@@ -31,6 +31,23 @@ def load_func(func_file)
end
end
+def load_env(file_path)
+ return unless File.exist?(file_path)
+
+ File.readlines(file_path).each do |line|
+ line = line.strip
+ next if line.empty? || line.start_with?('#')
+
+ key, *value = line.split('=', 2)
+ ENV[key.strip] = value.join('=').strip
+ end
+rescue StandardError
+end
+
+ENV['LLM_FUNCTIONS_DIR'] = Pathname.new(__dir__).join('..').expand_path.to_s
+
+load_env(Pathname.new(ENV['LLM_FUNCTIONS_DIR']).join('.env').to_s)
+
func_file, func_data = parse_argv
if ENV["LLM_FUNCTION_ACTION"] == "declarate"