@@ -17,7 +17,7 @@ def __init__(self, dendrite, config, subtensor, wallet: bt.wallet):
17
17
super ().__init__ (dendrite , config , subtensor , wallet , timeout = 75 )
18
18
self .streaming = True
19
19
self .query_type = "text"
20
- self .model = "gpt-4-1106-preview"
20
+ self .model = "gpt-4-1106-preview"
21
21
self .max_tokens = 4096
22
22
self .temperature = 0.0001
23
23
self .weight = 1
@@ -77,7 +77,7 @@ async def start_query(self, available_uids, metagraph) -> tuple[list, dict]:
77
77
query_tasks = []
78
78
uid_to_question = {}
79
79
# Randomly choose the provider based on specified probabilities
80
- providers = ["OpenAI" ] * 88 + ["Anthropic" ] * 2 + ["Gemini" ] * 0 + ["Claude" ] * 10
80
+ providers = ["OpenAI" ] * 70 + ["Anthropic" ] * 0 + ["Gemini" ] * 0 + ["Claude" ] * 30
81
81
self .provider = random .choice (providers )
82
82
83
83
if self .provider == "Anthropic" :
@@ -86,15 +86,17 @@ async def start_query(self, available_uids, metagraph) -> tuple[list, dict]:
86
86
# gemini models = ["gemini-pro"]
87
87
self .model = "anthropic.claude-v2:1"
88
88
elif self .provider == "OpenAI" :
89
- self .model = "gpt-4-1106-preview"
90
- # self.model = "gpt-3.5-turbo"
89
+ # self.model = "gpt-4-1106-preview"
90
+ self .model = "gpt-3.5-turbo"
91
91
92
92
elif self .provider == "Gemini" :
93
93
self .model = "gemini-pro"
94
94
95
95
elif self .provider == "Claude" :
96
- self .model = "claude-3-opus-20240229"
96
+ # self.model = "claude-3-opus-20240229"
97
97
# self.model = "claude-3-sonnet-20240229"
98
+ self .model = "claude-instant-1.2"
99
+
98
100
bt .logging .info (f"provider = { self .provider } \n model = { self .model } " )
99
101
for uid in available_uids :
100
102
prompt = await self .get_question (len (available_uids ))
@@ -116,7 +118,7 @@ async def start_query(self, available_uids, metagraph) -> tuple[list, dict]:
116
118
117
119
def should_i_score (self ):
118
120
random_number = random .random ()
119
- will_score_all = random_number < 1 / 12
121
+ will_score_all = random_number < 1 / 2
120
122
bt .logging .info (f"Random Number: { random_number } , Will score text responses: { will_score_all } " )
121
123
return will_score_all
122
124
0 commit comments