Merge pull request #50 from thedadams/add-cache-dir-opt

feat: add CacheDir option to GlobalOptions
gptscript-ai · Sep 3, 2024 · eb094a6 · eb094a6
2 parents e644c85 + 3d325fc
commit eb094a6
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/gptscript/opts.py b/gptscript/opts.py
@@ -9,12 +9,14 @@ def __init__(
             baseURL: str = "",
             defaultModelProvider: str = "",
             defaultModel: str = "",
+            cacheDir: str = "",
             env: Mapping[str, str] = None,
     ):
         self.APIKey = apiKey
         self.BaseURL = baseURL
         self.DefaultModel = defaultModel
         self.DefaultModelProvider = defaultModelProvider
+        self.CacheDir = cacheDir
         if env is None:
             env = os.environ
         env_list = [f"{k}={v}" for k, v in env.items()]
@@ -28,6 +30,7 @@ def merge(self, other: Self) -> Self:
         cp.BaseURL = other.BaseURL if other.BaseURL != "" else self.BaseURL
         cp.DefaultModel = other.DefaultModel if other.DefaultModel != "" else self.DefaultModel
         cp.DefaultModelProvider = other.DefaultModelProvider if other.DefaultModelProvider != "" else self.DefaultModelProvider
+        cp.CacheDir = other.CacheDir if other.CacheDir != "" else self.CacheDir
         cp.Env = (other.Env or []).extend(self.Env or [])
         return cp
 
@@ -61,9 +64,10 @@ def __init__(self,
                  apiKey: str = "",
                  baseURL: str = "",
                  defaultModelProvider: str = "",
-                 defaultModel: str = ""
+                 defaultModel: str = "",
+                 cacheDir: str = "",
                  ):
-        super().__init__(apiKey, baseURL, defaultModelProvider, defaultModel)
+        super().__init__(apiKey, baseURL, defaultModelProvider, defaultModel, cacheDir)
         self.input = input
         self.disableCache = disableCache
         self.subTool = subTool

diff --git a/tests/test_gptscript.py b/tests/test_gptscript.py
@@ -455,8 +455,8 @@ async def test_tool_chat(gptscript):
 async def test_file_chat(gptscript):
     inputs = [
         "List the 3 largest of the Great Lakes by volume.",
-        "For the second one in the list, what is the volume in cubic miles?",
-        "For the third one in the list, what is the total area in square miles?",
+        "What is the second largest?",
+        "What is the third one in the list?",
     ]
     expected_outputs = [
         "Lake Superior",