Commit ·
7ba4670
1
Parent(s): d59e3c0
- increasing max_try for request
Browse files- Messaging_system/LLM.py +4 -4
- Singeo_camp.csv +0 -0
Messaging_system/LLM.py
CHANGED
|
@@ -71,7 +71,7 @@ class LLM:
|
|
| 71 |
self.model = self.Core.model
|
| 72 |
|
| 73 |
# ==============================================================
|
| 74 |
-
def get_message_inference(self, prompt, instructions, max_retries=
|
| 75 |
"""
|
| 76 |
sending the prompt to openai LLM and get back the response
|
| 77 |
"""
|
|
@@ -155,7 +155,7 @@ class LLM:
|
|
| 155 |
return [], {}
|
| 156 |
|
| 157 |
# =========================================================================
|
| 158 |
-
def get_message_google(self, prompt, instructions, max_retries=
|
| 159 |
|
| 160 |
client = genai.Client(api_key=self.get_credential("Google_API"))
|
| 161 |
|
|
@@ -195,7 +195,7 @@ class LLM:
|
|
| 195 |
|
| 196 |
# =========================================================================
|
| 197 |
|
| 198 |
-
def get_message_openai(self, prompt, instructions, max_retries=
|
| 199 |
"""
|
| 200 |
sending the prompt to openai LLM and get back the response
|
| 201 |
"""
|
|
@@ -333,7 +333,7 @@ class LLM:
|
|
| 333 |
print("Max retries exceeded. Returning empty response.")
|
| 334 |
return [], {}
|
| 335 |
|
| 336 |
-
def get_message_claude(self, prompt, instructions, max_retries=
|
| 337 |
"""
|
| 338 |
send prompt to claude LLM and get back the response
|
| 339 |
:param prompt:
|
|
|
|
| 71 |
self.model = self.Core.model
|
| 72 |
|
| 73 |
# ==============================================================
|
| 74 |
+
def get_message_inference(self, prompt, instructions, max_retries=6):
|
| 75 |
"""
|
| 76 |
sending the prompt to openai LLM and get back the response
|
| 77 |
"""
|
|
|
|
| 155 |
return [], {}
|
| 156 |
|
| 157 |
# =========================================================================
|
| 158 |
+
def get_message_google(self, prompt, instructions, max_retries=6):
|
| 159 |
|
| 160 |
client = genai.Client(api_key=self.get_credential("Google_API"))
|
| 161 |
|
|
|
|
| 195 |
|
| 196 |
# =========================================================================
|
| 197 |
|
| 198 |
+
def get_message_openai(self, prompt, instructions, max_retries=5):
|
| 199 |
"""
|
| 200 |
sending the prompt to openai LLM and get back the response
|
| 201 |
"""
|
|
|
|
| 333 |
print("Max retries exceeded. Returning empty response.")
|
| 334 |
return [], {}
|
| 335 |
|
| 336 |
+
def get_message_claude(self, prompt, instructions, max_retries=6):
|
| 337 |
"""
|
| 338 |
send prompt to claude LLM and get back the response
|
| 339 |
:param prompt:
|
Singeo_camp.csv
CHANGED
|
The diff for this file is too large to render.
See raw diff
|
|
|