Skip to content

Commit cc438c8

Browse files
committed
add support for gpt-oss
1 parent 8277ca5 commit cc438c8

2 files changed

Lines changed: 12 additions & 0 deletions

File tree

setup.cfg

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -68,6 +68,7 @@ git_bob.prompt_handlers =
6868
blablador = git_bob._endpoints:prompt_blablador
6969
deepseek = git_bob._endpoints:prompt_deepseek
7070
e-infra_cz = git_bob._endpoints:prompt_e_infra_cz
71+
scads = git_bob._endpoints:prompt_scads
7172

7273
git_bob.triggers =
7374
review = git_bob._ai_github_utilities:review_pull_request

src/git_bob/_endpoints.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -140,6 +140,17 @@ def prompt_openai(message: str, model="gpt-4o-2024-08-06", image=None, max_accum
140140
# extract answer
141141
return result
142142

143+
def prompt_scads(message: str, model="openai/gpt-oss-120b", image=None, max_accumulated_responses=10, max_response_tokens=128000, base_url=None, api_key=None):
144+
import os
145+
if base_url is None:
146+
base_url = "hhttps://llm.scads.ai/v1"
147+
if api_key is None:
148+
api_key = os.environ.get("SCADS_API_KEY")
149+
model = model.replace("scads:", "")
150+
return prompt_openai(message, model=model, image=image, max_accumulated_responses=max_accumulated_responses, max_response_tokens=max_response_tokens, base_url=base_url, api_key=api_key)
151+
152+
153+
143154
def prompt_kisski(message: str, model=None, image=None, max_accumulated_responses=10, max_response_tokens=16384, base_url=None, api_key=None):
144155
import os
145156
if base_url is None:

0 commit comments

Comments
 (0)