Skip to content

Commit

Permalink
cleanup imports and dependencies
Browse files Browse the repository at this point in the history
  • Loading branch information
KastanDay committed Dec 14, 2023
1 parent 00834df commit ec6a2d4
Show file tree
Hide file tree
Showing 3 changed files with 6 additions and 4 deletions.
4 changes: 2 additions & 2 deletions ai_ta_backend/filtering_contexts.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,10 +5,10 @@

import openai
import ray
import replicate
import requests
from langchain import hub
from transformers import AutoTokenizer
# import replicate
# from transformers import AutoTokenizer

# load_dotenv(override=True)
# tokenizer = AutoTokenizer.from_pretrained("HuggingFaceH4/zephyr-7b-beta")
Expand Down
3 changes: 3 additions & 0 deletions ai_ta_backend/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,7 @@
from flask import Flask, Response, abort, jsonify, request
from flask_cors import CORS
from flask_executor import Executor
import ray
from sqlalchemy import JSON

from ai_ta_backend.nomic_logging import get_nomic_map, log_convo_to_nomic
Expand All @@ -23,6 +24,8 @@
# load API keys from globally-availabe .env file
load_dotenv()

ray.init()

@app.route('/')
def index() -> Response:
"""_summary_
Expand Down
3 changes: 1 addition & 2 deletions requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -55,5 +55,4 @@ unstructured==0.10.29 # causes huge ~5.3 GB of installs. Probbably from onnx: ht
xlrd # for excel ingest
newrelic
ray
langchainhub
replicate
langchainhub

0 comments on commit ec6a2d4

Please sign in to comment.