Commit ·
8c145de
1
Parent(s): a45b838
Switch to Groq llama-3.1-8b-instant via LiteLLM
Browse files- app.py +4 -4
- requirements.txt +1 -0
app.py
CHANGED
|
@@ -67,10 +67,10 @@ CRITICAL RULES:
|
|
| 67 |
"""
|
| 68 |
|
| 69 |
def build_agent():
|
| 70 |
-
model =
|
| 71 |
-
model_id="llama-3.1-8b-instant",
|
| 72 |
-
|
| 73 |
-
max_tokens=
|
| 74 |
temperature=0.1,
|
| 75 |
)
|
| 76 |
agent = CodeAgent(
|
|
|
|
| 67 |
"""
|
| 68 |
|
| 69 |
def build_agent():
|
| 70 |
+
model = LiteLLMModel(
|
| 71 |
+
model_id="groq/llama-3.1-8b-instant",
|
| 72 |
+
api_key=os.getenv("HF"),
|
| 73 |
+
max_tokens=1024,
|
| 74 |
temperature=0.1,
|
| 75 |
)
|
| 76 |
agent = CodeAgent(
|
requirements.txt
CHANGED
|
@@ -3,3 +3,4 @@ gradio
|
|
| 3 |
requests
|
| 4 |
pandas
|
| 5 |
duckduckgo-search
|
|
|
|
|
|
| 3 |
requests
|
| 4 |
pandas
|
| 5 |
duckduckgo-search
|
| 6 |
+
litellm
|