File size: 2,093 Bytes
c287182 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 | from pathlib import Path
p = Path("gateway.py")
text = p.read_text()
old = ''' else:
providers = [
("ollama", lambda: {"role":"assistant","content": llm_ollama(messages, model=model or "dolor3v_coder:turbo")}),
("groq", lambda: llm_groq(messages, model=model or "llama-3.3-70b-versatile")),
("cerebras", lambda: {"role":"assistant","content": llm_cerebras(messages, model=model or "gpt-oss-120b")}),
("glm", lambda: {"role":"assistant","content": llm_glm(messages, model=model or "glm-4.7-flash")}),
("openrouter", lambda: llm_openrouter(messages, model=model or "nvidia/nemotron-super-49b-v1:free")),
("openai", lambda: {"role":"assistant","content": llm_openai(messages, model=model or "gpt-4o-mini")}),
]'''
new = ''' else:
providers = [
("groq", lambda: llm_groq(
messages,
model=model or "llama-3.3-70b-versatile"
)),
("cerebras", lambda: {
"role":"assistant",
"content": llm_cerebras(
messages,
model=model or "gpt-oss-120b"
)
}),
("openrouter", lambda: llm_openrouter(
messages,
model=model or os.getenv(
"OPENROUTER_MODEL",
"qwen/qwen3-coder:free"
)
)),
]'''
if old not in text:
raise SystemExit("Provider block not found.")
text = text.replace(old, new)
old2 = ''' except Exception as e:
log(f"LLM {name} failed: {e}")
continue'''
new2 = ''' except Exception as e:
import traceback
traceback.print_exc()
print(f"[LLM ERROR] {name}: {repr(e)}")
log(f"LLM {name} failed: {repr(e)}")
continue'''
if old2 not in text:
raise SystemExit("Exception block not found.")
text = text.replace(old2, new2)
p.write_text(text)
print("gateway.py patched successfully.")
|