mirror of
https://github.com/leigest519/ScreenCoder.git
synced 2026-03-23 05:13:27 +01:00
Update utils.py
Change Bot class name for GPT and Qwen
This commit is contained in:
4
utils.py
4
utils.py
@@ -267,7 +267,7 @@ class Doubao(Bot):
|
|||||||
# img.show()
|
# img.show()
|
||||||
return response
|
return response
|
||||||
|
|
||||||
class Qwen_2_5_VL(Bot):
|
class Qwen(Bot):
|
||||||
def __init__(self, key_path, patience=3, model="qwen2.5-vl-32b-instruct") -> None:
|
def __init__(self, key_path, patience=3, model="qwen2.5-vl-32b-instruct") -> None:
|
||||||
super().__init__(key_path, patience)
|
super().__init__(key_path, patience)
|
||||||
self.client = OpenAI(api_key=self.key, base_url="https://dashscope.aliyuncs.com/compatible-mode/v1")
|
self.client = OpenAI(api_key=self.key, base_url="https://dashscope.aliyuncs.com/compatible-mode/v1")
|
||||||
@@ -306,7 +306,7 @@ class Qwen_2_5_VL(Bot):
|
|||||||
print("seed used: 42")
|
print("seed used: 42")
|
||||||
return response
|
return response
|
||||||
|
|
||||||
class GPT4(Bot):
|
class GPT(Bot):
|
||||||
def __init__(self, key_path, patience=3, model="gpt-4o") -> None:
|
def __init__(self, key_path, patience=3, model="gpt-4o") -> None:
|
||||||
super().__init__(key_path, patience)
|
super().__init__(key_path, patience)
|
||||||
self.client = OpenAI(api_key=self.key)
|
self.client = OpenAI(api_key=self.key)
|
||||||
|
|||||||
Reference in New Issue
Block a user