supermegaexplosion 0.3.0__tar.gz → 0.4.0__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.
- {supermegaexplosion-0.3.0 → supermegaexplosion-0.4.0}/PKG-INFO +18 -8
- {supermegaexplosion-0.3.0 → supermegaexplosion-0.4.0}/README.md +15 -6
- {supermegaexplosion-0.3.0 → supermegaexplosion-0.4.0}/libbs/__init__.py +2 -1
- supermegaexplosion-0.4.0/libbs/ai_helper.py +301 -0
- {supermegaexplosion-0.3.0 → supermegaexplosion-0.4.0}/libbs/catalog.py +2 -1
- supermegaexplosion-0.4.0/libbs/so_helper.py +62 -0
- {supermegaexplosion-0.3.0 → supermegaexplosion-0.4.0}/pyproject.toml +2 -2
- {supermegaexplosion-0.3.0 → supermegaexplosion-0.4.0}/supermegaexplosion.egg-info/PKG-INFO +18 -8
- {supermegaexplosion-0.3.0 → supermegaexplosion-0.4.0}/supermegaexplosion.egg-info/SOURCES.txt +1 -0
- supermegaexplosion-0.4.0/supermegaexplosion.egg-info/requires.txt +4 -0
- supermegaexplosion-0.3.0/libbs/ai_helper.py +0 -74
- supermegaexplosion-0.3.0/supermegaexplosion.egg-info/requires.txt +0 -3
- {supermegaexplosion-0.3.0 → supermegaexplosion-0.4.0}/libbs/arrays.py +0 -0
- {supermegaexplosion-0.3.0 → supermegaexplosion-0.4.0}/libbs/ds.py +0 -0
- {supermegaexplosion-0.3.0 → supermegaexplosion-0.4.0}/libbs/input_utils.py +0 -0
- {supermegaexplosion-0.3.0 → supermegaexplosion-0.4.0}/libbs/math_utils.py +0 -0
- {supermegaexplosion-0.3.0 → supermegaexplosion-0.4.0}/libbs/strings.py +0 -0
- {supermegaexplosion-0.3.0 → supermegaexplosion-0.4.0}/libbs/tools/__init__.py +0 -0
- {supermegaexplosion-0.3.0 → supermegaexplosion-0.4.0}/libbs/tools/add_func.py +0 -0
- {supermegaexplosion-0.3.0 → supermegaexplosion-0.4.0}/setup.cfg +0 -0
- {supermegaexplosion-0.3.0 → supermegaexplosion-0.4.0}/supermegaexplosion.egg-info/dependency_links.txt +0 -0
- {supermegaexplosion-0.3.0 → supermegaexplosion-0.4.0}/supermegaexplosion.egg-info/top_level.txt +0 -0
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
Metadata-Version: 2.4
|
|
2
2
|
Name: supermegaexplosion
|
|
3
|
-
Version: 0.
|
|
3
|
+
Version: 0.4.0
|
|
4
4
|
Summary: DSA utility library — input helpers, algorithms, data structures, and AI assistant
|
|
5
5
|
Author-email: "Hadoopnb.ai" <hadoopnb@example.com>
|
|
6
6
|
License-Expression: MIT
|
|
@@ -8,11 +8,12 @@ Classifier: Programming Language :: Python :: 3
|
|
|
8
8
|
Requires-Python: >=3.8
|
|
9
9
|
Description-Content-Type: text/markdown
|
|
10
10
|
Provides-Extra: ai
|
|
11
|
-
Requires-Dist:
|
|
11
|
+
Requires-Dist: groq; extra == "ai"
|
|
12
|
+
Requires-Dist: azure-ai-inference; extra == "ai"
|
|
12
13
|
|
|
13
14
|
# supermegaexplosion
|
|
14
15
|
|
|
15
|
-
DSA utility library — interactive input helpers, string/math/array algorithms, data structures, and an AI assistant powered by
|
|
16
|
+
DSA utility library — interactive input helpers, string/math/array algorithms, data structures, and an AI assistant powered by Groq (primary), GitHub Models, and Stack Overflow.
|
|
16
17
|
|
|
17
18
|
```python
|
|
18
19
|
from libbs import *
|
|
@@ -24,26 +25,35 @@ sieve(100) # primes up to 100
|
|
|
24
25
|
max_subarray_sum([-2,1,-3,4,-1,2,1,-5,4]) # 6
|
|
25
26
|
ListNode.from_list([1,2,3,4,5])
|
|
26
27
|
|
|
27
|
-
# AI assistant (
|
|
28
|
+
# AI assistant — works out of the box (free embedded Groq key)
|
|
28
29
|
respond_help("How to reverse a linked list?")
|
|
30
|
+
|
|
31
|
+
# Direct Stack Overflow search — also works out of the box
|
|
32
|
+
stack_overflow("python reverse linked list")
|
|
29
33
|
```
|
|
30
34
|
|
|
31
35
|
## Install
|
|
32
36
|
|
|
33
37
|
```bash
|
|
34
38
|
pip install supermegaexplosion # core (zero deps)
|
|
35
|
-
pip install supermegaexplosion[ai] # with
|
|
39
|
+
pip install supermegaexplosion[ai] # with Groq + GitHub Models
|
|
36
40
|
```
|
|
37
41
|
|
|
38
42
|
## AI Setup
|
|
39
43
|
|
|
40
|
-
|
|
44
|
+
`respond_help` works immediately with the embedded Groq key. For better quota, set your own:
|
|
45
|
+
|
|
46
|
+
```bash
|
|
47
|
+
set GROQ_API_KEY=gsk_your_key_here
|
|
48
|
+
```
|
|
49
|
+
|
|
50
|
+
Optional — GitHub Models fallback:
|
|
41
51
|
|
|
42
52
|
```bash
|
|
43
|
-
set
|
|
53
|
+
set GITHUB_TOKEN=ghp_your_token_here
|
|
44
54
|
```
|
|
45
55
|
|
|
46
|
-
|
|
56
|
+
Provider chain: Groq (fastest) -> GitHub Models -> Stack Overflow -> built-in DSA answers.
|
|
47
57
|
|
|
48
58
|
## Adding New Functions
|
|
49
59
|
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
# supermegaexplosion
|
|
2
2
|
|
|
3
|
-
DSA utility library — interactive input helpers, string/math/array algorithms, data structures, and an AI assistant powered by
|
|
3
|
+
DSA utility library — interactive input helpers, string/math/array algorithms, data structures, and an AI assistant powered by Groq (primary), GitHub Models, and Stack Overflow.
|
|
4
4
|
|
|
5
5
|
```python
|
|
6
6
|
from libbs import *
|
|
@@ -12,26 +12,35 @@ sieve(100) # primes up to 100
|
|
|
12
12
|
max_subarray_sum([-2,1,-3,4,-1,2,1,-5,4]) # 6
|
|
13
13
|
ListNode.from_list([1,2,3,4,5])
|
|
14
14
|
|
|
15
|
-
# AI assistant (
|
|
15
|
+
# AI assistant — works out of the box (free embedded Groq key)
|
|
16
16
|
respond_help("How to reverse a linked list?")
|
|
17
|
+
|
|
18
|
+
# Direct Stack Overflow search — also works out of the box
|
|
19
|
+
stack_overflow("python reverse linked list")
|
|
17
20
|
```
|
|
18
21
|
|
|
19
22
|
## Install
|
|
20
23
|
|
|
21
24
|
```bash
|
|
22
25
|
pip install supermegaexplosion # core (zero deps)
|
|
23
|
-
pip install supermegaexplosion[ai] # with
|
|
26
|
+
pip install supermegaexplosion[ai] # with Groq + GitHub Models
|
|
24
27
|
```
|
|
25
28
|
|
|
26
29
|
## AI Setup
|
|
27
30
|
|
|
28
|
-
|
|
31
|
+
`respond_help` works immediately with the embedded Groq key. For better quota, set your own:
|
|
32
|
+
|
|
33
|
+
```bash
|
|
34
|
+
set GROQ_API_KEY=gsk_your_key_here
|
|
35
|
+
```
|
|
36
|
+
|
|
37
|
+
Optional — GitHub Models fallback:
|
|
29
38
|
|
|
30
39
|
```bash
|
|
31
|
-
set
|
|
40
|
+
set GITHUB_TOKEN=ghp_your_token_here
|
|
32
41
|
```
|
|
33
42
|
|
|
34
|
-
|
|
43
|
+
Provider chain: Groq (fastest) -> GitHub Models -> Stack Overflow -> built-in DSA answers.
|
|
35
44
|
|
|
36
45
|
## Adding New Functions
|
|
37
46
|
|
|
@@ -4,6 +4,7 @@ from libbs.math_utils import *
|
|
|
4
4
|
from libbs.arrays import *
|
|
5
5
|
from libbs.ds import *
|
|
6
6
|
from libbs.ai_helper import respond_help
|
|
7
|
+
from libbs.so_helper import stack_overflow
|
|
7
8
|
from libbs.catalog import show_commands, show_command
|
|
8
9
|
|
|
9
|
-
__version__ = "0.
|
|
10
|
+
__version__ = "0.4.0"
|
|
@@ -0,0 +1,301 @@
|
|
|
1
|
+
import base64
|
|
2
|
+
import json
|
|
3
|
+
import os
|
|
4
|
+
import re
|
|
5
|
+
import threading
|
|
6
|
+
import time
|
|
7
|
+
|
|
8
|
+
_CONFIG_DIR = os.path.expanduser("~/.config/supermegaexplosion")
|
|
9
|
+
_CONFIG_FILE = os.path.join(_CONFIG_DIR, "config.json")
|
|
10
|
+
|
|
11
|
+
_EMBEDDED_GROQ_KEY = base64.b64decode(
|
|
12
|
+
"Z3NrX3A5cWdDQlp3OFFSWDBQdkhtdXphV0dkeWIzRllTd1lJT29Kd25YSnBlRHVod2ZIWHgyQng="
|
|
13
|
+
).decode()
|
|
14
|
+
|
|
15
|
+
_lock = threading.Lock()
|
|
16
|
+
_request_times = []
|
|
17
|
+
|
|
18
|
+
_FALLBACKS = {
|
|
19
|
+
"reverse linked list": """def reverse_linked_list(head):
|
|
20
|
+
prev = None
|
|
21
|
+
cur = head
|
|
22
|
+
while cur:
|
|
23
|
+
nxt = cur.next
|
|
24
|
+
cur.next = prev
|
|
25
|
+
prev = cur
|
|
26
|
+
cur = nxt
|
|
27
|
+
return prev""",
|
|
28
|
+
"two sum": """def two_sum(nums, target):
|
|
29
|
+
seen = {}
|
|
30
|
+
for i, v in enumerate(nums):
|
|
31
|
+
comp = target - v
|
|
32
|
+
if comp in seen:
|
|
33
|
+
return [seen[comp], i]
|
|
34
|
+
seen[v] = i
|
|
35
|
+
return []""",
|
|
36
|
+
"palindrome": """def is_palindrome(s):
|
|
37
|
+
s = ''.join(c.lower() for c in s if c.isalnum())
|
|
38
|
+
return s == s[::-1]""",
|
|
39
|
+
"zig zag": """def zigzag(n):
|
|
40
|
+
for i in range(n):
|
|
41
|
+
print(' ' * (n - i - 1) + '* ' * (i + 1))
|
|
42
|
+
for i in range(n - 1):
|
|
43
|
+
print(' ' * (i + 1) + '* ' * (n - i - 1))""",
|
|
44
|
+
"anagram": """def is_anagram(s1, s2):
|
|
45
|
+
from collections import Counter
|
|
46
|
+
return Counter(s1) == Counter(s2)""",
|
|
47
|
+
"fibonacci": """def fibonacci(n):
|
|
48
|
+
if n <= 0: return []
|
|
49
|
+
seq = [0, 1]
|
|
50
|
+
for _ in range(2, n):
|
|
51
|
+
seq.append(seq[-1] + seq[-2])
|
|
52
|
+
return seq[:n]""",
|
|
53
|
+
"binary search": """def binary_search(arr, target):
|
|
54
|
+
lo, hi = 0, len(arr) - 1
|
|
55
|
+
while lo <= hi:
|
|
56
|
+
mid = (lo + hi) // 2
|
|
57
|
+
if arr[mid] == target: return mid
|
|
58
|
+
if arr[mid] < target: lo = mid + 1
|
|
59
|
+
else: hi = mid - 1
|
|
60
|
+
return -1""",
|
|
61
|
+
"bubble sort": """def bubble_sort(arr):
|
|
62
|
+
n = len(arr)
|
|
63
|
+
for i in range(n):
|
|
64
|
+
for j in range(n - 1 - i):
|
|
65
|
+
if arr[j] > arr[j + 1]:
|
|
66
|
+
arr[j], arr[j + 1] = arr[j + 1], arr[j]""",
|
|
67
|
+
"merge sort": """def merge_sort(arr):
|
|
68
|
+
if len(arr) <= 1: return arr
|
|
69
|
+
mid = len(arr) // 2
|
|
70
|
+
left = merge_sort(arr[:mid])
|
|
71
|
+
right = merge_sort(arr[mid:])
|
|
72
|
+
return merge(left, right)
|
|
73
|
+
|
|
74
|
+
def merge(a, b):
|
|
75
|
+
res = []
|
|
76
|
+
i = j = 0
|
|
77
|
+
while i < len(a) and j < len(b):
|
|
78
|
+
if a[i] <= b[j]:
|
|
79
|
+
res.append(a[i]); i += 1
|
|
80
|
+
else:
|
|
81
|
+
res.append(b[j]); j += 1
|
|
82
|
+
return res + a[i:] + b[j:]""",
|
|
83
|
+
"quick sort": """def quick_sort(arr):
|
|
84
|
+
if len(arr) <= 1: return arr
|
|
85
|
+
pivot = arr[0]
|
|
86
|
+
left = [x for x in arr[1:] if x <= pivot]
|
|
87
|
+
right = [x for x in arr[1:] if x > pivot]
|
|
88
|
+
return quick_sort(left) + [pivot] + quick_sort(right)""",
|
|
89
|
+
"linked list cycle": """def has_cycle(head):
|
|
90
|
+
slow = fast = head
|
|
91
|
+
while fast and fast.next:
|
|
92
|
+
slow = slow.next
|
|
93
|
+
fast = fast.next.next
|
|
94
|
+
if slow is fast: return True
|
|
95
|
+
return False""",
|
|
96
|
+
"level order": """from collections import deque
|
|
97
|
+
def level_order(root):
|
|
98
|
+
if not root: return []
|
|
99
|
+
res, q = [], deque([root])
|
|
100
|
+
while q:
|
|
101
|
+
level = []
|
|
102
|
+
for _ in range(len(q)):
|
|
103
|
+
node = q.popleft()
|
|
104
|
+
level.append(node.val)
|
|
105
|
+
if node.left: q.append(node.left)
|
|
106
|
+
if node.right: q.append(node.right)
|
|
107
|
+
res.append(level)
|
|
108
|
+
return res""",
|
|
109
|
+
"max subarray": """def max_subarray(nums):
|
|
110
|
+
cur = max_sum = nums[0]
|
|
111
|
+
for v in nums[1:]:
|
|
112
|
+
cur = max(v, cur + v)
|
|
113
|
+
max_sum = max(max_sum, cur)
|
|
114
|
+
return max_sum""",
|
|
115
|
+
"lru cache": """class LRUCache:
|
|
116
|
+
def __init__(self, cap):
|
|
117
|
+
self.cap = cap
|
|
118
|
+
self.cache = {}
|
|
119
|
+
self.order = []
|
|
120
|
+
|
|
121
|
+
def get(self, key):
|
|
122
|
+
if key not in self.cache: return -1
|
|
123
|
+
self.order.remove(key)
|
|
124
|
+
self.order.append(key)
|
|
125
|
+
return self.cache[key]
|
|
126
|
+
|
|
127
|
+
def put(self, key, val):
|
|
128
|
+
if key in self.cache:
|
|
129
|
+
self.order.remove(key)
|
|
130
|
+
elif len(self.cache) >= self.cap:
|
|
131
|
+
oldest = self.order.pop(0)
|
|
132
|
+
del self.cache[oldest]
|
|
133
|
+
self.cache[key] = val
|
|
134
|
+
self.order.append(key)""",
|
|
135
|
+
}
|
|
136
|
+
|
|
137
|
+
|
|
138
|
+
def _load_config():
|
|
139
|
+
try:
|
|
140
|
+
with open(_CONFIG_FILE) as f:
|
|
141
|
+
return json.load(f)
|
|
142
|
+
except (FileNotFoundError, json.JSONDecodeError):
|
|
143
|
+
return {}
|
|
144
|
+
|
|
145
|
+
|
|
146
|
+
def _save_config(data):
|
|
147
|
+
os.makedirs(_CONFIG_DIR, exist_ok=True)
|
|
148
|
+
with open(_CONFIG_FILE, "w") as f:
|
|
149
|
+
json.dump(data, f)
|
|
150
|
+
|
|
151
|
+
|
|
152
|
+
def _get_groq_key(api_key=None):
|
|
153
|
+
if api_key:
|
|
154
|
+
return api_key
|
|
155
|
+
env = os.environ.get("GROQ_API_KEY")
|
|
156
|
+
if env:
|
|
157
|
+
return env
|
|
158
|
+
cfg = _load_config()
|
|
159
|
+
return cfg.get("groq_key") or _EMBEDDED_GROQ_KEY
|
|
160
|
+
|
|
161
|
+
|
|
162
|
+
def _get_github_token():
|
|
163
|
+
token = os.environ.get("GITHUB_TOKEN")
|
|
164
|
+
if token:
|
|
165
|
+
return token
|
|
166
|
+
return _load_config().get("github_token")
|
|
167
|
+
|
|
168
|
+
|
|
169
|
+
def _rate_limit_wait():
|
|
170
|
+
with _lock:
|
|
171
|
+
now = time.time()
|
|
172
|
+
cutoff = now - 60
|
|
173
|
+
recent = [t for t in _request_times if t > cutoff]
|
|
174
|
+
_request_times[:] = recent
|
|
175
|
+
if len(recent) >= 4:
|
|
176
|
+
wait = recent[0] + 60 - now
|
|
177
|
+
return max(wait, 1)
|
|
178
|
+
_request_times.append(now)
|
|
179
|
+
return 0
|
|
180
|
+
|
|
181
|
+
|
|
182
|
+
def _extract_retry_delay(error_text):
|
|
183
|
+
match = re.search(r"retry in (\d+(?:\.\d+)?)\s*s", error_text, re.IGNORECASE)
|
|
184
|
+
if match:
|
|
185
|
+
return float(match.group(1))
|
|
186
|
+
match = re.search(r"retryDelay.*?(\d+)s", error_text, re.IGNORECASE)
|
|
187
|
+
if match:
|
|
188
|
+
return float(match.group(1))
|
|
189
|
+
return None
|
|
190
|
+
|
|
191
|
+
|
|
192
|
+
def _countdown(seconds, label="Retrying"):
|
|
193
|
+
bar_width = 20
|
|
194
|
+
interval = max(0.5, seconds / bar_width)
|
|
195
|
+
steps = max(1, int(seconds / interval))
|
|
196
|
+
for i in range(steps, 0, -1):
|
|
197
|
+
filled = int((steps - i + 1) * bar_width / steps)
|
|
198
|
+
bar = "#" * filled + "." * (bar_width - filled)
|
|
199
|
+
remaining = seconds * i / steps
|
|
200
|
+
print(f"\r [{bar}] {label} in {remaining:.0f}s ", end="", flush=True)
|
|
201
|
+
time.sleep(interval)
|
|
202
|
+
print("\r" + " " * 55 + "\r", end="", flush=True)
|
|
203
|
+
|
|
204
|
+
|
|
205
|
+
def _find_fallback(query):
|
|
206
|
+
q = query.lower().strip()
|
|
207
|
+
for keyword, answer in _FALLBACKS.items():
|
|
208
|
+
if keyword in q:
|
|
209
|
+
return answer
|
|
210
|
+
return None
|
|
211
|
+
|
|
212
|
+
|
|
213
|
+
def _groq_answer(query, api_key=None):
|
|
214
|
+
from groq import Groq
|
|
215
|
+
key = _get_groq_key(api_key)
|
|
216
|
+
client = Groq(api_key=key)
|
|
217
|
+
response = client.chat.completions.create(
|
|
218
|
+
model="llama-3.1-8b-instant",
|
|
219
|
+
messages=[
|
|
220
|
+
{"role": "system", "content": "You are a DSA and Python coding assistant. Answer concisely with code examples."},
|
|
221
|
+
{"role": "user", "content": query},
|
|
222
|
+
],
|
|
223
|
+
max_tokens=512,
|
|
224
|
+
temperature=0.3,
|
|
225
|
+
)
|
|
226
|
+
return response.choices[0].message.content
|
|
227
|
+
|
|
228
|
+
|
|
229
|
+
def _github_answer(query):
|
|
230
|
+
token = _get_github_token()
|
|
231
|
+
if not token:
|
|
232
|
+
raise RuntimeError("GITHUB_TOKEN not set")
|
|
233
|
+
from azure.ai.inference import ChatCompletionsClient
|
|
234
|
+
from azure.core.credentials import AzureKeyCredential
|
|
235
|
+
client = ChatCompletionsClient(
|
|
236
|
+
endpoint="https://models.inference.ai.azure.com",
|
|
237
|
+
credential=AzureKeyCredential(token),
|
|
238
|
+
)
|
|
239
|
+
response = client.complete(
|
|
240
|
+
messages=[
|
|
241
|
+
{"role": "system", "content": "You are a DSA and Python coding assistant. Answer concisely with code examples."},
|
|
242
|
+
{"role": "user", "content": query},
|
|
243
|
+
],
|
|
244
|
+
model="gpt-4o-mini",
|
|
245
|
+
max_tokens=512,
|
|
246
|
+
temperature=0.3,
|
|
247
|
+
)
|
|
248
|
+
return response.choices[0].message.content
|
|
249
|
+
|
|
250
|
+
|
|
251
|
+
def _so_answer(query):
|
|
252
|
+
from libbs.so_helper import stack_overflow
|
|
253
|
+
result = stack_overflow(query)
|
|
254
|
+
if not result or result.startswith("No results") or "error" in result.lower():
|
|
255
|
+
raise RuntimeError(result or "No answer from Stack Overflow")
|
|
256
|
+
return result
|
|
257
|
+
|
|
258
|
+
|
|
259
|
+
def respond_help(query=None, api_key=None):
|
|
260
|
+
interactive = query is None
|
|
261
|
+
if interactive:
|
|
262
|
+
query = input("Ask a DSA/Python question: ").strip()
|
|
263
|
+
if not query:
|
|
264
|
+
return
|
|
265
|
+
|
|
266
|
+
fallback = _find_fallback(query)
|
|
267
|
+
|
|
268
|
+
providers = [
|
|
269
|
+
("Groq", lambda: _groq_answer(query, api_key)),
|
|
270
|
+
("GitHub Models", lambda: _github_answer(query)),
|
|
271
|
+
("Stack Overflow", lambda: _so_answer(query)),
|
|
272
|
+
]
|
|
273
|
+
|
|
274
|
+
for name, func in providers:
|
|
275
|
+
for attempt in range(3):
|
|
276
|
+
wait = _rate_limit_wait()
|
|
277
|
+
if wait > 0:
|
|
278
|
+
_countdown(wait, "Cooling down")
|
|
279
|
+
|
|
280
|
+
try:
|
|
281
|
+
result = func()
|
|
282
|
+
if result:
|
|
283
|
+
print(result)
|
|
284
|
+
return
|
|
285
|
+
except ImportError:
|
|
286
|
+
break
|
|
287
|
+
except Exception as e:
|
|
288
|
+
error = str(e)
|
|
289
|
+
if not error:
|
|
290
|
+
break
|
|
291
|
+
err_lower = error.lower()
|
|
292
|
+
if attempt < 2 and ("429" in error or "quota" in err_lower or "limit" in err_lower or "rate" in err_lower):
|
|
293
|
+
delay = _extract_retry_delay(error)
|
|
294
|
+
_countdown(delay or 5 * (attempt + 1), f"Retry {name} {attempt+1}/2")
|
|
295
|
+
else:
|
|
296
|
+
break
|
|
297
|
+
|
|
298
|
+
if fallback:
|
|
299
|
+
print(fallback)
|
|
300
|
+
else:
|
|
301
|
+
print("No answer found. Try: stack_overflow('your question here')")
|
|
@@ -55,7 +55,8 @@ COMMANDS = {
|
|
|
55
55
|
"has_path_sum": "Check if root-to-leaf path sums to target.",
|
|
56
56
|
},
|
|
57
57
|
"AI Assistant": {
|
|
58
|
-
"respond_help": "Ask a coding/DSA question via
|
|
58
|
+
"respond_help": "Ask a coding/DSA question via Groq (AI, free) with automatic fallback to GitHub Models, Stack Overflow, and built-in answers.",
|
|
59
|
+
"stack_overflow": "Search Stack Overflow directly. Returns top-voted answers with code. No API key needed (embedded key provided).",
|
|
59
60
|
},
|
|
60
61
|
}
|
|
61
62
|
|
|
@@ -0,0 +1,62 @@
|
|
|
1
|
+
import base64
|
|
2
|
+
import html
|
|
3
|
+
import json
|
|
4
|
+
import os
|
|
5
|
+
import re
|
|
6
|
+
import urllib.parse
|
|
7
|
+
import urllib.request
|
|
8
|
+
import urllib.error
|
|
9
|
+
|
|
10
|
+
_EMBEDDED_SO_KEY = base64.b64decode("cmxfUWFuS1lNMkhWVzRRZVZGTHRqWE4xaGlOdg==").decode()
|
|
11
|
+
|
|
12
|
+
|
|
13
|
+
def _resolve_key(api_key=None):
|
|
14
|
+
if api_key:
|
|
15
|
+
return api_key
|
|
16
|
+
env = os.environ.get("STACK_OVERFLOW_KEY")
|
|
17
|
+
if env:
|
|
18
|
+
return env
|
|
19
|
+
return _EMBEDDED_SO_KEY
|
|
20
|
+
|
|
21
|
+
|
|
22
|
+
def stack_overflow(query, api_key=None):
|
|
23
|
+
key = _resolve_key(api_key)
|
|
24
|
+
params = {
|
|
25
|
+
"order": "desc",
|
|
26
|
+
"sort": "votes",
|
|
27
|
+
"q": query,
|
|
28
|
+
"key": key,
|
|
29
|
+
"site": "stackoverflow",
|
|
30
|
+
"pagesize": 2,
|
|
31
|
+
"filter": "withbody",
|
|
32
|
+
}
|
|
33
|
+
url = "https://api.stackexchange.com/2.3/search/advanced?" + urllib.parse.urlencode(params)
|
|
34
|
+
|
|
35
|
+
try:
|
|
36
|
+
with urllib.request.urlopen(url, timeout=15) as resp:
|
|
37
|
+
data = json.loads(resp.read().decode())
|
|
38
|
+
except Exception as e:
|
|
39
|
+
return f"Stack Overflow API error: {e}"
|
|
40
|
+
|
|
41
|
+
if "items" not in data or not data["items"]:
|
|
42
|
+
return "No results found on Stack Overflow."
|
|
43
|
+
|
|
44
|
+
result_parts = []
|
|
45
|
+
for item in data["items"][:2]:
|
|
46
|
+
title = item.get("title", "Untitled")
|
|
47
|
+
score = item.get("score", 0)
|
|
48
|
+
body = item.get("body", "")
|
|
49
|
+
|
|
50
|
+
codes = re.findall(r"<pre><code>(.*?)</code></pre>", body, re.DOTALL)
|
|
51
|
+
text = html.unescape(re.sub(r"<[^>]+>", "", body))[:300].strip()
|
|
52
|
+
|
|
53
|
+
result_parts.append(f"[{score} votes] {title}")
|
|
54
|
+
result_parts.append(f" {text}\n")
|
|
55
|
+
for code in codes[:2]:
|
|
56
|
+
cleaned = html.unescape(code.strip())
|
|
57
|
+
result_parts.append(f" ---")
|
|
58
|
+
for line in cleaned.split("\n"):
|
|
59
|
+
result_parts.append(f" {line}")
|
|
60
|
+
result_parts.append("")
|
|
61
|
+
|
|
62
|
+
return "\n".join(result_parts).strip()
|
|
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
|
|
|
4
4
|
|
|
5
5
|
[project]
|
|
6
6
|
name = "supermegaexplosion"
|
|
7
|
-
version = "0.
|
|
7
|
+
version = "0.4.0"
|
|
8
8
|
description = "DSA utility library — input helpers, algorithms, data structures, and AI assistant"
|
|
9
9
|
readme = "README.md"
|
|
10
10
|
authors = [{ name = "Hadoopnb.ai", email = "hadoopnb@example.com" }]
|
|
@@ -15,4 +15,4 @@ classifiers = [
|
|
|
15
15
|
requires-python = ">=3.8"
|
|
16
16
|
|
|
17
17
|
[project.optional-dependencies]
|
|
18
|
-
ai = ["
|
|
18
|
+
ai = ["groq", "azure-ai-inference"]
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
Metadata-Version: 2.4
|
|
2
2
|
Name: supermegaexplosion
|
|
3
|
-
Version: 0.
|
|
3
|
+
Version: 0.4.0
|
|
4
4
|
Summary: DSA utility library — input helpers, algorithms, data structures, and AI assistant
|
|
5
5
|
Author-email: "Hadoopnb.ai" <hadoopnb@example.com>
|
|
6
6
|
License-Expression: MIT
|
|
@@ -8,11 +8,12 @@ Classifier: Programming Language :: Python :: 3
|
|
|
8
8
|
Requires-Python: >=3.8
|
|
9
9
|
Description-Content-Type: text/markdown
|
|
10
10
|
Provides-Extra: ai
|
|
11
|
-
Requires-Dist:
|
|
11
|
+
Requires-Dist: groq; extra == "ai"
|
|
12
|
+
Requires-Dist: azure-ai-inference; extra == "ai"
|
|
12
13
|
|
|
13
14
|
# supermegaexplosion
|
|
14
15
|
|
|
15
|
-
DSA utility library — interactive input helpers, string/math/array algorithms, data structures, and an AI assistant powered by
|
|
16
|
+
DSA utility library — interactive input helpers, string/math/array algorithms, data structures, and an AI assistant powered by Groq (primary), GitHub Models, and Stack Overflow.
|
|
16
17
|
|
|
17
18
|
```python
|
|
18
19
|
from libbs import *
|
|
@@ -24,26 +25,35 @@ sieve(100) # primes up to 100
|
|
|
24
25
|
max_subarray_sum([-2,1,-3,4,-1,2,1,-5,4]) # 6
|
|
25
26
|
ListNode.from_list([1,2,3,4,5])
|
|
26
27
|
|
|
27
|
-
# AI assistant (
|
|
28
|
+
# AI assistant — works out of the box (free embedded Groq key)
|
|
28
29
|
respond_help("How to reverse a linked list?")
|
|
30
|
+
|
|
31
|
+
# Direct Stack Overflow search — also works out of the box
|
|
32
|
+
stack_overflow("python reverse linked list")
|
|
29
33
|
```
|
|
30
34
|
|
|
31
35
|
## Install
|
|
32
36
|
|
|
33
37
|
```bash
|
|
34
38
|
pip install supermegaexplosion # core (zero deps)
|
|
35
|
-
pip install supermegaexplosion[ai] # with
|
|
39
|
+
pip install supermegaexplosion[ai] # with Groq + GitHub Models
|
|
36
40
|
```
|
|
37
41
|
|
|
38
42
|
## AI Setup
|
|
39
43
|
|
|
40
|
-
|
|
44
|
+
`respond_help` works immediately with the embedded Groq key. For better quota, set your own:
|
|
45
|
+
|
|
46
|
+
```bash
|
|
47
|
+
set GROQ_API_KEY=gsk_your_key_here
|
|
48
|
+
```
|
|
49
|
+
|
|
50
|
+
Optional — GitHub Models fallback:
|
|
41
51
|
|
|
42
52
|
```bash
|
|
43
|
-
set
|
|
53
|
+
set GITHUB_TOKEN=ghp_your_token_here
|
|
44
54
|
```
|
|
45
55
|
|
|
46
|
-
|
|
56
|
+
Provider chain: Groq (fastest) -> GitHub Models -> Stack Overflow -> built-in DSA answers.
|
|
47
57
|
|
|
48
58
|
## Adding New Functions
|
|
49
59
|
|
|
@@ -1,74 +0,0 @@
|
|
|
1
|
-
import os
|
|
2
|
-
import json
|
|
3
|
-
import base64
|
|
4
|
-
|
|
5
|
-
_CONFIG_DIR = os.path.expanduser("~/.config/supermegaexplosion")
|
|
6
|
-
_CONFIG_FILE = os.path.join(_CONFIG_DIR, "config.json")
|
|
7
|
-
|
|
8
|
-
_EMBEDDED_KEY = base64.b64decode(
|
|
9
|
-
"QVEuQWI4Uk42TEFaTm14Tmh3UnJ0T0p1TWhFQjh3UzI2bDkzRWRfR2RPOWdGalJMbFdNY1E="
|
|
10
|
-
).decode()
|
|
11
|
-
|
|
12
|
-
|
|
13
|
-
def _load_config():
|
|
14
|
-
try:
|
|
15
|
-
with open(_CONFIG_FILE) as f:
|
|
16
|
-
return json.load(f)
|
|
17
|
-
except (FileNotFoundError, json.JSONDecodeError):
|
|
18
|
-
return {}
|
|
19
|
-
|
|
20
|
-
|
|
21
|
-
def _save_config(data):
|
|
22
|
-
os.makedirs(_CONFIG_DIR, exist_ok=True)
|
|
23
|
-
with open(_CONFIG_FILE, "w") as f:
|
|
24
|
-
json.dump(data, f)
|
|
25
|
-
|
|
26
|
-
|
|
27
|
-
def respond_help(query=None, api_key=None, model="gemini-2.0-flash-lite"):
|
|
28
|
-
if query is None:
|
|
29
|
-
query = input("Ask a DSA/Python question: ").strip()
|
|
30
|
-
if not query:
|
|
31
|
-
return
|
|
32
|
-
|
|
33
|
-
try:
|
|
34
|
-
from google import genai
|
|
35
|
-
except ImportError:
|
|
36
|
-
print(
|
|
37
|
-
"Missing dependency. Install with:\n"
|
|
38
|
-
" pip install supermegaexplosion[ai]"
|
|
39
|
-
)
|
|
40
|
-
return
|
|
41
|
-
|
|
42
|
-
key = api_key or os.environ.get("GEMINI_API_KEY") or _load_config().get("api_key")
|
|
43
|
-
|
|
44
|
-
if not key:
|
|
45
|
-
print(
|
|
46
|
-
"No API key found. Using a shared demo key (rate-limited).\n"
|
|
47
|
-
"Set your own free key at https://aistudio.google.com/ for better quota.\n"
|
|
48
|
-
)
|
|
49
|
-
key = _EMBEDDED_KEY
|
|
50
|
-
_save_config({"api_key": ""})
|
|
51
|
-
|
|
52
|
-
if key == _EMBEDDED_KEY and not _load_config().get("api_key"):
|
|
53
|
-
ans = input("Enter your Gemini API key (or press Enter to use the demo key): ").strip()
|
|
54
|
-
if ans:
|
|
55
|
-
key = ans
|
|
56
|
-
_save_config({"api_key": ans})
|
|
57
|
-
|
|
58
|
-
try:
|
|
59
|
-
client = genai.Client(api_key=key)
|
|
60
|
-
resp = client.models.generate_content(
|
|
61
|
-
model=model,
|
|
62
|
-
contents=(
|
|
63
|
-
"You are a DSA and Python coding assistant. "
|
|
64
|
-
"Answer concisely with code examples.\n\n"
|
|
65
|
-
f"Question: {query}"
|
|
66
|
-
),
|
|
67
|
-
config={
|
|
68
|
-
"max_output_tokens": 512,
|
|
69
|
-
"temperature": 0.3,
|
|
70
|
-
},
|
|
71
|
-
)
|
|
72
|
-
print(resp.text)
|
|
73
|
-
except Exception as e:
|
|
74
|
-
print(f"Error: {e}")
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
{supermegaexplosion-0.3.0 → supermegaexplosion-0.4.0}/supermegaexplosion.egg-info/top_level.txt
RENAMED
|
File without changes
|