rechka/fish-station
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Refactor imports and update AI review function

Browse source
This commit is contained in:
Vigers Ray 2025-11-15 03:21:22 +01:00 • committed by GitHub
parent ab4da0ef91
commit f19225b6f7
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 9 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
Tools/_sunrise/ai_review.py
View file

@ -3,8 +3,7 @@ import sys
import json
import subprocess
from azure.ai.inference import ChatCompletionsClient
from azure.ai.inference.models import SystemMessage
from azure.ai.inference.models import UserMessage
from azure.ai.inference.models import SystemMessage, UserMessage
from azure.core.credentials import AzureKeyCredential
def get_diff():
@ -26,9 +25,13 @@ def get_pr_body():
return f.read().strip()
def run_ai_review(gdd, diff):
endpoint = "https://models.github.ai/inference"
model_name = "ai21-labs/AI21-Jamba-1.5-Large"
token = os.environ["GITHUB_TOKEN"]
client = ChatCompletionsClient(
endpoint="https://models.github.ai/inference",
credential=AzureKeyCredential(os.environ["GITHUB_TOKEN"]),
endpoint=endpoint,
credential=AzureKeyCredential(token),
)
title = get_pr_title()
@ -65,10 +68,10 @@ def run_ai_review(gdd, diff):
SystemMessage(content="You are an expert senior game designer."),
UserMessage(content=prompt),
],
model="ai21-labs/AI21-Jamba-1.5-Large",
temperature=0.3,
top_p=1.0,
max_tokens=5000,
top_p=1.0
model=model_name
)
return response.choices[0].message.content