gmsg 0.1.6__tar.gz → 0.1.8__tar.gz
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- {gmsg-0.1.6 → gmsg-0.1.8}/PKG-INFO +1 -1
- {gmsg-0.1.6 → gmsg-0.1.8}/gmsg/gmsg.py +4 -4
- {gmsg-0.1.6 → gmsg-0.1.8}/pyproject.toml +1 -1
- {gmsg-0.1.6 → gmsg-0.1.8}/LICENSE +0 -0
- {gmsg-0.1.6 → gmsg-0.1.8}/README.md +0 -0
- {gmsg-0.1.6 → gmsg-0.1.8}/gmsg/__init__.py +0 -0
- {gmsg-0.1.6 → gmsg-0.1.8}/gmsg/api_key.py +0 -0
|
@@ -4,8 +4,8 @@ import tempfile
|
|
|
4
4
|
import subprocess
|
|
5
5
|
from google import genai
|
|
6
6
|
from pathlib import Path
|
|
7
|
-
|
|
8
|
-
from api_key import get_or_set_api_key
|
|
7
|
+
from .api_key import get_or_set_api_key
|
|
8
|
+
# from api_key import get_or_set_api_key
|
|
9
9
|
|
|
10
10
|
GREEN = "\033[92m"
|
|
11
11
|
RED = "\033[91m"
|
|
@@ -43,7 +43,7 @@ def make_query(diff: str) -> str:
|
|
|
43
43
|
def trigger_query(query: str, api_key: str) -> str:
|
|
44
44
|
try:
|
|
45
45
|
client = genai.Client(api_key=api_key)
|
|
46
|
-
msg = client.models.generate_content(model="gemini-2.
|
|
46
|
+
msg = client.models.generate_content(model="gemini-2.5-flash",
|
|
47
47
|
contents=query)
|
|
48
48
|
return msg.text.strip()
|
|
49
49
|
except genai.errors.ClientError as error:
|
|
@@ -52,7 +52,7 @@ def trigger_query(query: str, api_key: str) -> str:
|
|
|
52
52
|
is_success=False)
|
|
53
53
|
sys.exit(1)
|
|
54
54
|
else:
|
|
55
|
-
|
|
55
|
+
printt(str(error), is_success=False)
|
|
56
56
|
sys.exit(1)
|
|
57
57
|
|
|
58
58
|
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|