added replicate to requirements.txt

UIUC-Chatbot · Dec 5, 2023 · 3e5a553 · 3e5a553
1 parent 0f72b73
commit 3e5a553
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/ai_ta_backend/filtering_contexts.py b/ai_ta_backend/filtering_contexts.py
@@ -87,7 +87,7 @@ def run_anyscale(prompt):
           api_base = "https://api.endpoints.anyscale.com/v1",
           api_key=os.environ["ANYSCALE_ENDPOINT_TOKEN"],
           # model="meta-llama/Llama-2-70b-chat-hf",
-          engine="mistralai/Mistral-7B-Instruct-v0.1",
+          model="mistralai/Mistral-7B-Instruct-v0.1",
           messages=[{"role": "system", "content": "You are a helpful assistant."},
                     {"role": "user", "content": prompt}],
           temperature=0.3, 

diff --git a/requirements.txt b/requirements.txt
@@ -55,4 +55,5 @@ unstructured==0.10.29 # causes huge ~5.3 GB of installs. Probbably from onnx: ht
 xlrd # for excel ingest
 newrelic
 ray
-langchainhub
+langchainhub
+replicate