chore: add credential check param field

Signed-off-by: Donnie Adams <[email protected]>

Author: thedadams

gptscript/credentials.py

@@ -22,6 +22,7 @@ def __init__(self,
                  ephemeral: bool = False,
                  expiresAt: datetime = None,
                  refreshToken: str = "",
+                 checkParam: str = "",
                  **kwargs,
                  ):
         self.context = context
@@ -31,6 +32,7 @@ def __init__(self,
         self.ephemeral = ephemeral
         self.expiresAt = expiresAt
         self.refreshToken = refreshToken
+        self.checkParam = checkParam
 
         if self.env is None:
             self.env = {}
@@ -56,6 +58,7 @@ def to_json(self):
             "env": self.env,
             "ephemeral": self.ephemeral,
             "refreshToken": self.refreshToken,
+            "checkParam": self.checkParam,
         }
 
         if datetime_str != "":
@@ -95,4 +98,5 @@ def to_credential(c) -> Credential:
         ephemeral=c.get("ephemeral", False),
         expiresAt=expiresAt,
         refreshToken=c["refreshToken"],
+        checkParam=c.get("checkParam", "")
     )

tests/test_gptscript.py

@@ -742,7 +742,8 @@ async def test_credentials(gptscript):
     name = "test-" + str(os.urandom(4).hex())
     now = datetime.now()
     res = await gptscript.create_credential(
-        Credential(toolName=name, env={"TEST": "test"}, expiresAt=now + timedelta(seconds=5)))
+        Credential(toolName=name, env={"TEST": "test"}, expiresAt=now + timedelta(seconds=5),
+                   checkParam="my-check-param"))
     assert not res.startswith("an error occurred"), "Unexpected error creating credential: " + res
 
     sleep(5)
@@ -757,6 +758,7 @@ async def test_credentials(gptscript):
     res = await gptscript.reveal_credential(name=name)
     assert not str(res).startswith("an error occurred"), "Unexpected error revealing credential: " + res
     assert res.env["TEST"] == "test", "Unexpected credential value: " + str(res)
+    assert res.checkParam == "my-check-param", "Unexpected credential value: " + str(res)
 
     res = await gptscript.delete_credential(name=name)
     assert not res.startswith("an error occurred"), "Unexpected error deleting credential: " + res