From 6cd555dc48af8772b64cd047669fa1771a333f3a Mon Sep 17 00:00:00 2001
From: Zion Leonahenahe Basque <zion@zionbasque.com>
Date: Wed, 7 Aug 2024 10:12:12 -0700
Subject: [PATCH] Fix broken Decompilation interface in libbs (#50)

* Fix broken Decompilation interface in libbs

* bump vers

* Bump min Py version to 3.10

* Bump min Py version to 3.10
---
 .github/workflows/tests.yml            | 2 +-
 dailalib/__init__.py                   | 2 +-
 dailalib/api/litellm/prompts/prompt.py | 3 +--
 setup.cfg                              | 8 ++++----
 4 files changed, 7 insertions(+), 8 deletions(-)

diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
index 30d0c45..c6f947d 100644
--- a/.github/workflows/tests.yml
+++ b/.github/workflows/tests.yml
@@ -16,7 +16,7 @@ jobs:
     strategy:
       matrix:
         os: [ubuntu-latest]
-        python-version: [3.8]
+        python-version: [3.10.14]
 
     steps:
       - uses: actions/checkout@v2
diff --git a/dailalib/__init__.py b/dailalib/__init__.py
index 49c59eb..58ea681 100644
--- a/dailalib/__init__.py
+++ b/dailalib/__init__.py
@@ -1,4 +1,4 @@
-__version__ = "3.4.1"
+__version__ = "3.4.2"
 
 from .api import AIAPI, LiteLLMAIAPI
 from libbs.api import DecompilerInterface
diff --git a/dailalib/api/litellm/prompts/prompt.py b/dailalib/api/litellm/prompts/prompt.py
index 1667545..56ba5b5 100644
--- a/dailalib/api/litellm/prompts/prompt.py
+++ b/dailalib/api/litellm/prompts/prompt.py
@@ -1,5 +1,4 @@
 import json
-import typing
 import re
 from typing import Optional, Union, Dict, Callable
 import textwrap
@@ -48,7 +47,7 @@ def query_model(self, *args, function=None, dec_text=None, use_dec=True, **kwarg
             raise Exception("api must be set before querying!")
 
         @AIAPI.requires_function
-        def _query_model(ai_api=self.ai_api, function=function, dec_text=dec_text, use_dec=use_dec) -> Union[Dict, str]:
+        def _query_model(ai_api=self.ai_api, function=function, dec_text=dec_text, **_kwargs) -> Union[Dict, str]:
             if not ai_api:
                 return {}
 
diff --git a/setup.cfg b/setup.cfg
index 2db598e..162cb78 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -5,7 +5,7 @@ url = https://github.com/mahaloz/DAILA
 classifiers =
     License :: OSI Approved :: BSD License
     Programming Language :: Python :: 3
-    Programming Language :: Python :: 3.8
+    Programming Language :: Python :: 3.10
 license = BSD 2 Clause
 license_files = LICENSE
 description = The Decompiler Artificial Intelligence Language Assistant (DAILA) is a tool for adding AI to decompilers.
@@ -17,9 +17,9 @@ install_requires =
     litellm
     tiktoken
     Jinja2
-    libbs>=1.17.1
+    libbs>=1.18.1
 
-python_requires = >= 3.8
+python_requires = >= 3.10
 include_package_data = True
 packages = find:
 
@@ -29,4 +29,4 @@ console_scripts =
 
 [options.extras_require]
 full =
-    varbert>=2.1.0
+    varbert>=2.1.1