Spaces:
Running
Running
Remove non working llms for now...
Browse files
ring.py
CHANGED
|
@@ -8,6 +8,8 @@ from openworm_ai.utils.llms import GENERAL_QUERY_LIMITED_PROMPT_TEMPLATE
|
|
| 8 |
|
| 9 |
from openworm_ai.utils.llms import get_llm as get_llm_via_openworm_ai
|
| 10 |
|
|
|
|
|
|
|
| 11 |
from openworm_ai.utils.llms import get_gemini_api_key
|
| 12 |
|
| 13 |
|
|
@@ -32,7 +34,7 @@ LLM_GEMINI = "gemini-1.5-pro"
|
|
| 32 |
LLM_CLAUDE2 = "Claude2.1"
|
| 33 |
|
| 34 |
|
| 35 |
-
PREF_ORDER_LLMS = [LLM_GEMINI] + OPENWORM_AI_LLMS + [LLM_LLAMA2, LLM_CLAUDE2]
|
| 36 |
|
| 37 |
|
| 38 |
def get_llamaapi_key():
|
|
|
|
| 8 |
|
| 9 |
from openworm_ai.utils.llms import get_llm as get_llm_via_openworm_ai
|
| 10 |
|
| 11 |
+
from openworm_ai.utils.llms import requires_openai_key
|
| 12 |
+
from openworm_ai.utils.llms import get_openai_api_key
|
| 13 |
from openworm_ai.utils.llms import get_gemini_api_key
|
| 14 |
|
| 15 |
|
|
|
|
| 34 |
LLM_CLAUDE2 = "Claude2.1"
|
| 35 |
|
| 36 |
|
| 37 |
+
PREF_ORDER_LLMS = [LLM_GEMINI] + OPENWORM_AI_LLMS # + [LLM_LLAMA2, LLM_CLAUDE2]
|
| 38 |
|
| 39 |
|
| 40 |
def get_llamaapi_key():
|