debug print result; kimi temp=1.0 compat
This commit is contained in:
@@ -189,6 +189,7 @@ if __name__ == '__main__':
|
|||||||
try:
|
try:
|
||||||
while 'done' not in (item := dq.get(timeout=120)): pass
|
while 'done' not in (item := dq.get(timeout=120)): pass
|
||||||
result = item['done']
|
result = item['done']
|
||||||
|
print(result)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
if once: raise
|
if once: raise
|
||||||
print(f'[Reflect] drain error: {e}'); result = f'[ERROR] {e}'
|
print(f'[Reflect] drain error: {e}'); result = f'[ERROR] {e}'
|
||||||
|
|||||||
@@ -84,6 +84,7 @@ class ClaudeSession:
|
|||||||
return result[::-1] or messages[-2:]
|
return result[::-1] or messages[-2:]
|
||||||
def raw_ask(self, messages, model=None, temperature=0.5, max_tokens=6144):
|
def raw_ask(self, messages, model=None, temperature=0.5, max_tokens=6144):
|
||||||
model = model or self.default_model
|
model = model or self.default_model
|
||||||
|
if 'kimi' in model.lower(): temperature = 1.0 # kimi only accepts temp 1.0
|
||||||
headers = {"x-api-key": self.api_key, "Content-Type": "application/json", "anthropic-version": "2023-06-01"}
|
headers = {"x-api-key": self.api_key, "Content-Type": "application/json", "anthropic-version": "2023-06-01"}
|
||||||
payload = {"model": model, "messages": messages, "temperature": temperature, "max_tokens": max_tokens, "stream": True}
|
payload = {"model": model, "messages": messages, "temperature": temperature, "max_tokens": max_tokens, "stream": True}
|
||||||
try:
|
try:
|
||||||
@@ -168,6 +169,7 @@ class LLMSession:
|
|||||||
|
|
||||||
def raw_ask(self, messages, model=None, temperature=0.5):
|
def raw_ask(self, messages, model=None, temperature=0.5):
|
||||||
if model is None: model = self.default_model
|
if model is None: model = self.default_model
|
||||||
|
if 'kimi' in model.lower(): temperature = 1.0 # kimi only accepts temp 1.0
|
||||||
headers = {"Authorization": f"Bearer {self.api_key}", "Content-Type": "application/json", "Accept": "text/event-stream"}
|
headers = {"Authorization": f"Bearer {self.api_key}", "Content-Type": "application/json", "Accept": "text/event-stream"}
|
||||||
if self.api_mode == "responses":
|
if self.api_mode == "responses":
|
||||||
url = auto_make_url(self.api_base, "responses")
|
url = auto_make_url(self.api_base, "responses")
|
||||||
|
|||||||
Reference in New Issue
Block a user