fix order

Josh-XT · Jun 27, 2024 · 64d7572 · 64d7572
1 parent e7b0de3
commit 64d7572
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/agixt/providers/google.py b/agixt/providers/google.py
@@ -46,6 +46,7 @@ def services():
         return ["llm", "tts", "vision"]
 
     async def inference(self, prompt, tokens: int = 0, images: list = []):
+        new_max_tokens = int(self.MAX_TOKENS) - tokens
         if not self.GOOGLE_API_KEY or self.GOOGLE_API_KEY == "None":
             return "Please set your Google API key in the Agent Management page."
         try:
@@ -57,7 +58,6 @@ async def inference(self, prompt, tokens: int = 0, images: list = []):
                 model_name=self.AI_MODEL if not images else "gemini-pro-vision",
                 generation_config=generation_config,
             )
-            new_max_tokens = int(self.MAX_TOKENS) - tokens
             new_prompt = []
             if images:
                 for image in images:

diff --git a/agixt/version b/agixt/version
@@ -1 +1 @@
-v1.6.0
+v1.6.1