This commit is contained in:
Alex
2023-03-02 14:14:47 +00:00
parent 873dfb2703
commit e35961911c
2 changed files with 26 additions and 4 deletions

View File

@@ -6,7 +6,8 @@ import dotenv
import requests
from flask import Flask, request, render_template
from langchain import FAISS
from langchain import OpenAI, VectorDBQA, HuggingFaceHub, Cohere
from langchain.llms import OpenAIChat
from langchain import VectorDBQA, HuggingFaceHub, Cohere
from langchain.chains.question_answering import load_qa_chain
from langchain.embeddings import OpenAIEmbeddings, HuggingFaceHubEmbeddings, CohereEmbeddings, HuggingFaceInstructEmbeddings
from langchain.prompts import PromptTemplate
@@ -115,7 +116,7 @@ def api_answer():
c_prompt = PromptTemplate(input_variables=["summaries", "question"], template=template, template_format="jinja2")
if llm_choice == "openai":
llm = OpenAI(openai_api_key=api_key, temperature=0)
llm = OpenAIChat(openai_api_key=api_key, temperature=0)
elif llm_choice == "manifest":
llm = ManifestWrapper(client=manifest, llm_kwargs={"temperature": 0.001, "max_tokens": 2048})
elif llm_choice == "huggingface":

View File

@@ -1,6 +1,9 @@
aiodns==3.0.0
aiohttp==3.8.3
aiohttp-retry==2.8.3
aiosignal==1.3.1
alabaster==0.7.13
aleph-alpha-client==2.16.0
anyio==3.6.2
argilla==1.3.0
async-timeout==4.0.2
@@ -8,14 +11,20 @@ attrs==22.2.0
Babel==2.11.0
backoff==2.2.1
blobfile==2.0.1
boto3==1.26.82
botocore==1.29.82
cffi==1.15.1
charset-normalizer==2.1.1
click==8.1.3
cohere==3.4.0
dataclasses-json==0.5.7
decorator==5.1.1
deeplake==3.2.12
Deprecated==1.2.13
dill==0.3.6
docutils==0.19
docx2txt==0.8
entrypoints==0.4
escodegen==1.0.10
esprima==4.0.1
esutils==1.0.1
@@ -29,14 +38,17 @@ gunicorn==20.1.0
h11==0.14.0
httpcore==0.16.3
httpx==0.23.3
hub==3.0.1
huggingface-hub==0.12.0
humbug==0.2.8
idna==3.4
imagesize==1.4.1
itsdangerous==2.1.2
javalang==0.13.0
Jinja2==3.1.2
jmespath==1.0.1
joblib==1.2.0
langchain==0.0.81
langchain==0.0.98
lxml==4.9.2
manifest-ml==0.1.1
MarkupSafe==2.1.2
@@ -44,18 +56,26 @@ marshmallow==3.19.0
marshmallow-enum==1.5.1
monotonic==1.6
multidict==6.0.4
multiprocess==0.70.14
mypy-extensions==0.4.3
nltk==3.8.1
numcodecs==0.11.0
numpy==1.23.5
openai==0.26.4
openai==0.27.0
openpyxl==3.1.1
packaging==23.0
pandas==1.5.3
pathos==0.3.0
Pillow==9.4.0
pox==0.3.2
ppft==1.7.6.6
py==1.11.0
pycares==4.3.0
pycparser==2.21
pycryptodomex==3.17
pydantic==1.10.4
Pygments==2.14.0
PyJWT==2.6.0
PyPDF2==3.0.1
python-dateutil==2.8.2
python-docx==0.8.11
@@ -69,6 +89,7 @@ regex==2022.10.31
requests==2.28.2
retry==0.9.2
rfc3986==1.5.0
s3transfer==0.6.0
scikit-learn==1.2.1
scipy==1.10.0
sentence-transformers==2.2.2