From 0c6b609c261cb6f586668626d860dc6754725794 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Fri, 7 Jun 2024 21:58:46 +0800
Subject: refactor: demo_tool.* inspect LLM_* env vars (#33)

---
 tools/demo_tool.py | 16 +++++++---------
 1 file changed, 7 insertions(+), 9 deletions(-)

(limited to 'tools/demo_tool.py')

diff --git a/tools/demo_tool.py b/tools/demo_tool.py
index 5eb4c09..d53cd0b 100644
--- a/tools/demo_tool.py
+++ b/tools/demo_tool.py
@@ -1,6 +1,6 @@
+import os
 from typing import List, Literal, Optional
 
-
 def run(
     boolean: bool,
     string: str,
@@ -22,11 +22,9 @@ def run(
         string_optional: Define a optional string property
         array_optional: Define a optional string array property
     """
-    print(f"boolean: {boolean}")
-    print(f"string: {string}")
-    print(f"string_enum: {string_enum}")
-    print(f"integer: {integer}")
-    print(f"number: {number}")
-    print(f"array: {array}")
-    print(f"string_optional: {string_optional}")
-    print(f"array_optional: {array_optional}")
+    for key, value in locals().items():
+        print(f"{key}: {value}")
+
+    for key, value in os.environ.items():
+        if key.startswith("LLM_"):
+            print(f"{key}: {value}")
\ No newline at end of file
-- 
cgit v1.2.3