JKilpatrick commited on
Commit
e88ca0c
β€’
1 Parent(s): b8b9f57

initial commit

Browse files
Files changed (5) hide show
  1. .gitignore +160 -0
  2. Dockerfile +11 -0
  3. app.py +116 -0
  4. chainlit.md +14 -0
  5. requirements.txt +12 -0
.gitignore ADDED
@@ -0,0 +1,160 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Byte-compiled / optimized / DLL files
2
+ __pycache__/
3
+ *.py[cod]
4
+ *$py.class
5
+
6
+ # C extensions
7
+ *.so
8
+
9
+ # Distribution / packaging
10
+ .Python
11
+ build/
12
+ develop-eggs/
13
+ dist/
14
+ downloads/
15
+ eggs/
16
+ .eggs/
17
+ lib/
18
+ lib64/
19
+ parts/
20
+ sdist/
21
+ var/
22
+ wheels/
23
+ share/python-wheels/
24
+ *.egg-info/
25
+ .installed.cfg
26
+ *.egg
27
+ MANIFEST
28
+
29
+ # PyInstaller
30
+ # Usually these files are written by a python script from a template
31
+ # before PyInstaller builds the exe, so as to inject date/other infos into it.
32
+ *.manifest
33
+ *.spec
34
+
35
+ # Installer logs
36
+ pip-log.txt
37
+ pip-delete-this-directory.txt
38
+
39
+ # Unit test / coverage reports
40
+ htmlcov/
41
+ .tox/
42
+ .nox/
43
+ .coverage
44
+ .coverage.*
45
+ .cache
46
+ nosetests.xml
47
+ coverage.xml
48
+ *.cover
49
+ *.py,cover
50
+ .hypothesis/
51
+ .pytest_cache/
52
+ cover/
53
+
54
+ # Translations
55
+ *.mo
56
+ *.pot
57
+
58
+ # Django stuff:
59
+ *.log
60
+ local_settings.py
61
+ db.sqlite3
62
+ db.sqlite3-journal
63
+
64
+ # Flask stuff:
65
+ instance/
66
+ .webassets-cache
67
+
68
+ # Scrapy stuff:
69
+ .scrapy
70
+
71
+ # Sphinx documentation
72
+ docs/_build/
73
+
74
+ # PyBuilder
75
+ .pybuilder/
76
+ target/
77
+
78
+ # Jupyter Notebook
79
+ .ipynb_checkpoints
80
+
81
+ # IPython
82
+ profile_default/
83
+ ipython_config.py
84
+
85
+ # pyenv
86
+ # For a library or package, you might want to ignore these files since the code is
87
+ # intended to run in multiple environments; otherwise, check them in:
88
+ # .python-version
89
+
90
+ # pipenv
91
+ # According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
92
+ # However, in case of collaboration, if having platform-specific dependencies or dependencies
93
+ # having no cross-platform support, pipenv may install dependencies that don't work, or not
94
+ # install all needed dependencies.
95
+ #Pipfile.lock
96
+
97
+ # poetry
98
+ # Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
99
+ # This is especially recommended for binary packages to ensure reproducibility, and is more
100
+ # commonly ignored for libraries.
101
+ # https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
102
+ #poetry.lock
103
+
104
+ # pdm
105
+ # Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
106
+ #pdm.lock
107
+ # pdm stores project-wide configurations in .pdm.toml, but it is recommended to not include it
108
+ # in version control.
109
+ # https://pdm.fming.dev/#use-with-ide
110
+ .pdm.toml
111
+
112
+ # PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
113
+ __pypackages__/
114
+
115
+ # Celery stuff
116
+ celerybeat-schedule
117
+ celerybeat.pid
118
+
119
+ # SageMath parsed files
120
+ *.sage.py
121
+
122
+ # Environments
123
+ .env
124
+ .venv
125
+ env/
126
+ venv/
127
+ ENV/
128
+ env.bak/
129
+ venv.bak/
130
+
131
+ # Spyder project settings
132
+ .spyderproject
133
+ .spyproject
134
+
135
+ # Rope project settings
136
+ .ropeproject
137
+
138
+ # mkdocs documentation
139
+ /site
140
+
141
+ # mypy
142
+ .mypy_cache/
143
+ .dmypy.json
144
+ dmypy.json
145
+
146
+ # Pyre type checker
147
+ .pyre/
148
+
149
+ # pytype static type analyzer
150
+ .pytype/
151
+
152
+ # Cython debug symbols
153
+ cython_debug/
154
+
155
+ # PyCharm
156
+ # JetBrains specific template is maintained in a separate JetBrains.gitignore that can
157
+ # be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
158
+ # and can be added to the global gitignore or merged into this file. For a more nuclear
159
+ # option (not recommended) you can uncomment the following to ignore the entire idea folder.
160
+ #.idea/
Dockerfile ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ FROM python:3.9
2
+ RUN useradd -m -u 1000 user
3
+ USER user
4
+ ENV HOME=/home/user \
5
+ PATH=/home/user/.local/bin:$PATH
6
+ WORKDIR $HOME/app
7
+ COPY --chown=user . $HOME/app
8
+ COPY ./requirements.txt ~/app/requirements.txt
9
+ RUN pip install -r requirements.txt
10
+ COPY . .
11
+ CMD ["chainlit", "run", "app.py", "--port", "7860"]
app.py ADDED
@@ -0,0 +1,116 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ import os
3
+ from langchain.embeddings.openai import OpenAIEmbeddings
4
+ from langchain.embeddings import CacheBackedEmbeddings
5
+ from langchain.storage import LocalFileStore
6
+
7
+ import chainlit as cl
8
+ from chainlit.playground.providers import ChatOpenAI
9
+ from dotenv import load_dotenv
10
+ load_dotenv()
11
+
12
+ from langchain.chat_models import ChatOpenAI
13
+ from langchain.prompts import ChatPromptTemplate
14
+ from langchain.schema.runnable import RunnablePassthrough
15
+ from langchain.embeddings.openai import OpenAIEmbeddings
16
+ from langchain.embeddings import CacheBackedEmbeddings
17
+ from langchain.storage import LocalFileStore
18
+ from langchain.vectorstores import Pinecone
19
+
20
+ from operator import itemgetter
21
+ import pinecone
22
+
23
+ # =============================================================================
24
+ # Retrieval Chain
25
+ # =============================================================================
26
+ def load_llm():
27
+ llm = ChatOpenAI(
28
+ model='gpt-3.5-turbo',
29
+ temperature=0.0,
30
+ )
31
+ return llm
32
+
33
+
34
+ def load_vectorstore():
35
+
36
+ pinecone.init(
37
+ api_key=os.getenv('PINECONE_API_KEY'),
38
+ environment=os.getenv('PINECONE_ENV')
39
+ )
40
+
41
+ index = pinecone.GRPCIndex("youtube-index")
42
+ store = LocalFileStore("./cache/")
43
+ core_embeddings_model = OpenAIEmbeddings()
44
+
45
+ embedder = CacheBackedEmbeddings.from_bytes_store(
46
+ core_embeddings_model,
47
+ store,
48
+ namespace=core_embeddings_model.model
49
+ )
50
+
51
+ text_field = "text"
52
+
53
+ index = pinecone.Index("youtube-index")
54
+
55
+ vectorstore = Pinecone(
56
+ index,
57
+ embedder,
58
+ text_field
59
+ )
60
+
61
+ return vectorstore
62
+
63
+
64
+ def qa_chain():
65
+
66
+ vectorstore = load_vectorstore()
67
+
68
+ llm = load_llm()
69
+ retriever = vectorstore.as_retriever()
70
+
71
+ template = """You are a helpful assistant that answers questions on the provided context, if its not answered within the context respond with I dont know.
72
+ Additionally, the context includes a specific integer formatted as <int>, representing a timestamp. In your response, include this integer as a citation, formatted as a YouTube video link: "https://www.youtube.com/watch?v=[video_id]&t=<int>s" and text of link be the title of video.
73
+
74
+
75
+ ### CONTEXT
76
+ {context}
77
+
78
+ ### QUESTION
79
+ {question}
80
+ """
81
+
82
+ prompt = ChatPromptTemplate.from_template(template)
83
+
84
+ retrieval_augmented_qa_chain = (
85
+ {"context": itemgetter("question") | retriever,
86
+ "question": itemgetter("question")
87
+ }
88
+ | RunnablePassthrough.assign(
89
+ context=itemgetter("context")
90
+ )
91
+ | {
92
+ "response": prompt | llm,
93
+ "context": itemgetter("context"),
94
+ }
95
+ )
96
+
97
+ return retrieval_augmented_qa_chain
98
+
99
+ # =============================================================================
100
+ # Chainlit
101
+ # =============================================================================
102
+ @cl.on_chat_start
103
+ async def on_chat_start():
104
+ chain = qa_chain()
105
+ cl.user_session.set("chain", chain)
106
+ msg=cl.Message(content="What is your question about AI Makerspace?")
107
+ await msg.send()
108
+
109
+ @cl.on_message
110
+ async def on_message(message: cl.Message):
111
+ chain=cl.user_session.get("chain")
112
+ res = chain.invoke({"question" : message.content})
113
+
114
+ answer = res['response'].content
115
+
116
+ await cl.Message(content=answer).send()
chainlit.md ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Welcome to Chainlit! πŸš€πŸ€–
2
+
3
+ Hi there, Developer! πŸ‘‹ We're excited to have you on board. Chainlit is a powerful tool designed to help you prototype, debug and share applications built on top of LLMs.
4
+
5
+ ## Useful Links πŸ”—
6
+
7
+ - **Documentation:** Get started with our comprehensive [Chainlit Documentation](https://docs.chainlit.io) πŸ“š
8
+ - **Discord Community:** Join our friendly [Chainlit Discord](https://discord.gg/k73SQ3FyUh) to ask questions, share your projects, and connect with other developers! πŸ’¬
9
+
10
+ We can't wait to see what you create with Chainlit! Happy coding! πŸ’»πŸ˜Š
11
+
12
+ ## Welcome screen
13
+
14
+ To modify the welcome screen, edit the `chainlit.md` file at the root of your project. If you do not want a welcome screen, just leave this file empty.
requirements.txt ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ openai==0.27.2
2
+ pandas==1.4.2
3
+ requests==2.29.0
4
+ youtube_transcript_api==0.6.0
5
+
6
+ chainlit==0.7.700
7
+ cohere==4.37
8
+ openai==1.3.5
9
+ tiktoken==0.5.1
10
+ python-dotenv==1.0.0
11
+ pinecone-client==2.2.4
12
+ langchain==0.0.350