Skip to content

Commit 4862a6c

Browse files
committed
Revert "Add support for Azure, OpenAI, Palm, Anthropic, Cohere, Replicate Models - using litellm (#54)"
This reverts commit 08166a3, reversing changes made to e656aaf.
1 parent 34125c7 commit 4862a6c

File tree

2 files changed

+1
-4
lines changed

2 files changed

+1
-4
lines changed

Diff for: app/gpt.py

+1-3
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,6 @@
55
import random
66
import uuid
77
import openai
8-
import litellm
98
from pathlib import Path
109
from llama_index import ServiceContext, GPTVectorStoreIndex, LLMPredictor, RssReader, SimpleDirectoryReader, StorageContext, load_index_from_storage
1110
from llama_index.readers.schema.base import Document
@@ -105,12 +104,11 @@ def get_index_name_from_file(file: str):
105104
file_md5 = file_md5_with_extension.split('.')[0]
106105
return file_md5
107106

108-
# use any of the models specified here: https://litellm.readthedocs.io/en/latest/supported/
109107
def get_answer_from_chatGPT(messages):
110108
dialog_messages = format_dialog_messages(messages)
111109
logging.info('=====> Use chatGPT to answer!')
112110
logging.info(dialog_messages)
113-
completion = litellm.completion(
111+
completion = openai.ChatCompletion.create(
114112
model="gpt-3.5-turbo",
115113
messages=[{"role": "user", "content": dialog_messages}]
116114
)

Diff for: requirements.txt

-1
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,6 @@ Flask==2.2.3
22
gunicorn==20.1.0
33
llama-index==0.6.9
44
langchain==0.0.154
5-
litellm==0.1.232
65
httpx==0.23.3
76
requests==2.28.2
87
openai==0.27.1

0 commit comments

Comments
 (0)