Upload folder using huggingface_hub
Browse files- .github/workflows/update_space.yml +28 -0
- .gitignore +2 -0
- .gradio/certificate.pem +31 -0
- LICENSE +21 -0
- README.md +9 -7
- __pycache__/utilis.cpython-311.pyc +0 -0
- app.py +30 -0
- requirements.txt +183 -0
- utilis.py +65 -0
.github/workflows/update_space.yml
ADDED
@@ -0,0 +1,28 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
name: Run Python script
|
2 |
+
|
3 |
+
on:
|
4 |
+
push:
|
5 |
+
branches:
|
6 |
+
- main
|
7 |
+
|
8 |
+
jobs:
|
9 |
+
build:
|
10 |
+
runs-on: ubuntu-latest
|
11 |
+
|
12 |
+
steps:
|
13 |
+
- name: Checkout
|
14 |
+
uses: actions/checkout@v2
|
15 |
+
|
16 |
+
- name: Set up Python
|
17 |
+
uses: actions/setup-python@v2
|
18 |
+
with:
|
19 |
+
python-version: '3.9'
|
20 |
+
|
21 |
+
- name: Install Gradio
|
22 |
+
run: python -m pip install gradio
|
23 |
+
|
24 |
+
- name: Log in to Hugging Face
|
25 |
+
run: python -c 'import huggingface_hub; huggingface_hub.login(token="${{ secrets.hf_token }}")'
|
26 |
+
|
27 |
+
- name: Deploy to Spaces
|
28 |
+
run: gradio deploy
|
.gitignore
ADDED
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
1 |
+
venv
|
2 |
+
.env
|
.gradio/certificate.pem
ADDED
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
-----BEGIN CERTIFICATE-----
|
2 |
+
MIIFazCCA1OgAwIBAgIRAIIQz7DSQONZRGPgu2OCiwAwDQYJKoZIhvcNAQELBQAw
|
3 |
+
TzELMAkGA1UEBhMCVVMxKTAnBgNVBAoTIEludGVybmV0IFNlY3VyaXR5IFJlc2Vh
|
4 |
+
cmNoIEdyb3VwMRUwEwYDVQQDEwxJU1JHIFJvb3QgWDEwHhcNMTUwNjA0MTEwNDM4
|
5 |
+
WhcNMzUwNjA0MTEwNDM4WjBPMQswCQYDVQQGEwJVUzEpMCcGA1UEChMgSW50ZXJu
|
6 |
+
ZXQgU2VjdXJpdHkgUmVzZWFyY2ggR3JvdXAxFTATBgNVBAMTDElTUkcgUm9vdCBY
|
7 |
+
MTCCAiIwDQYJKoZIhvcNAQEBBQADggIPADCCAgoCggIBAK3oJHP0FDfzm54rVygc
|
8 |
+
h77ct984kIxuPOZXoHj3dcKi/vVqbvYATyjb3miGbESTtrFj/RQSa78f0uoxmyF+
|
9 |
+
0TM8ukj13Xnfs7j/EvEhmkvBioZxaUpmZmyPfjxwv60pIgbz5MDmgK7iS4+3mX6U
|
10 |
+
A5/TR5d8mUgjU+g4rk8Kb4Mu0UlXjIB0ttov0DiNewNwIRt18jA8+o+u3dpjq+sW
|
11 |
+
T8KOEUt+zwvo/7V3LvSye0rgTBIlDHCNAymg4VMk7BPZ7hm/ELNKjD+Jo2FR3qyH
|
12 |
+
B5T0Y3HsLuJvW5iB4YlcNHlsdu87kGJ55tukmi8mxdAQ4Q7e2RCOFvu396j3x+UC
|
13 |
+
B5iPNgiV5+I3lg02dZ77DnKxHZu8A/lJBdiB3QW0KtZB6awBdpUKD9jf1b0SHzUv
|
14 |
+
KBds0pjBqAlkd25HN7rOrFleaJ1/ctaJxQZBKT5ZPt0m9STJEadao0xAH0ahmbWn
|
15 |
+
OlFuhjuefXKnEgV4We0+UXgVCwOPjdAvBbI+e0ocS3MFEvzG6uBQE3xDk3SzynTn
|
16 |
+
jh8BCNAw1FtxNrQHusEwMFxIt4I7mKZ9YIqioymCzLq9gwQbooMDQaHWBfEbwrbw
|
17 |
+
qHyGO0aoSCqI3Haadr8faqU9GY/rOPNk3sgrDQoo//fb4hVC1CLQJ13hef4Y53CI
|
18 |
+
rU7m2Ys6xt0nUW7/vGT1M0NPAgMBAAGjQjBAMA4GA1UdDwEB/wQEAwIBBjAPBgNV
|
19 |
+
HRMBAf8EBTADAQH/MB0GA1UdDgQWBBR5tFnme7bl5AFzgAiIyBpY9umbbjANBgkq
|
20 |
+
hkiG9w0BAQsFAAOCAgEAVR9YqbyyqFDQDLHYGmkgJykIrGF1XIpu+ILlaS/V9lZL
|
21 |
+
ubhzEFnTIZd+50xx+7LSYK05qAvqFyFWhfFQDlnrzuBZ6brJFe+GnY+EgPbk6ZGQ
|
22 |
+
3BebYhtF8GaV0nxvwuo77x/Py9auJ/GpsMiu/X1+mvoiBOv/2X/qkSsisRcOj/KK
|
23 |
+
NFtY2PwByVS5uCbMiogziUwthDyC3+6WVwW6LLv3xLfHTjuCvjHIInNzktHCgKQ5
|
24 |
+
ORAzI4JMPJ+GslWYHb4phowim57iaztXOoJwTdwJx4nLCgdNbOhdjsnvzqvHu7Ur
|
25 |
+
TkXWStAmzOVyyghqpZXjFaH3pO3JLF+l+/+sKAIuvtd7u+Nxe5AW0wdeRlN8NwdC
|
26 |
+
jNPElpzVmbUq4JUagEiuTDkHzsxHpFKVK7q4+63SM1N95R1NbdWhscdCb+ZAJzVc
|
27 |
+
oyi3B43njTOQ5yOf+1CceWxG1bQVs5ZufpsMljq4Ui0/1lvh+wjChP4kqKOJ2qxq
|
28 |
+
4RgqsahDYVvTH9w7jXbyLeiNdd8XM2w9U/t7y0Ff/9yi0GE44Za4rF2LN9d11TPA
|
29 |
+
mRGunUHBcnWEvgJBQl9nJEiU0Zsnvgc/ubhPgXRR4Xq37Z0j4r7g1SgEEzwxA57d
|
30 |
+
emyPxgcYxn/eR44/KJ4EBs+lVDR3veyJm+kXQ99b21/+jh5Xos1AnX5iItreGCc=
|
31 |
+
-----END CERTIFICATE-----
|
LICENSE
ADDED
@@ -0,0 +1,21 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
MIT License
|
2 |
+
|
3 |
+
Copyright (c) 2024 Mohd Umar Javaid
|
4 |
+
|
5 |
+
Permission is hereby granted, free of charge, to any person obtaining a copy
|
6 |
+
of this software and associated documentation files (the "Software"), to deal
|
7 |
+
in the Software without restriction, including without limitation the rights
|
8 |
+
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
9 |
+
copies of the Software, and to permit persons to whom the Software is
|
10 |
+
furnished to do so, subject to the following conditions:
|
11 |
+
|
12 |
+
The above copyright notice and this permission notice shall be included in all
|
13 |
+
copies or substantial portions of the Software.
|
14 |
+
|
15 |
+
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
16 |
+
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
17 |
+
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
18 |
+
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
19 |
+
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
20 |
+
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
21 |
+
SOFTWARE.
|
README.md
CHANGED
@@ -1,12 +1,14 @@
|
|
1 |
---
|
2 |
-
title:
|
3 |
-
|
4 |
-
colorFrom: yellow
|
5 |
-
colorTo: gray
|
6 |
sdk: gradio
|
7 |
sdk_version: 5.15.0
|
8 |
-
app_file: app.py
|
9 |
-
pinned: false
|
10 |
---
|
|
|
|
|
11 |
|
12 |
-
|
|
|
|
|
|
|
|
|
|
1 |
---
|
2 |
+
title: shayari
|
3 |
+
app_file: app.py
|
|
|
|
|
4 |
sdk: gradio
|
5 |
sdk_version: 5.15.0
|
|
|
|
|
6 |
---
|
7 |
+
# Shayari Search App
|
8 |
+
### Semantic search for poetry
|
9 |
|
10 |
+
# Tech stack used
|
11 |
+
- Langchain
|
12 |
+
- Mistral LLM
|
13 |
+
- Together Embeddings
|
14 |
+
- Pinecone Vector DB
|
__pycache__/utilis.cpython-311.pyc
ADDED
Binary file (3.51 kB). View file
|
|
app.py
ADDED
@@ -0,0 +1,30 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from langchain_mistralai import ChatMistralAI
|
2 |
+
import os
|
3 |
+
from dotenv import load_dotenv
|
4 |
+
import gradio as gr
|
5 |
+
from utilis import retrieve, llm
|
6 |
+
|
7 |
+
load_dotenv()
|
8 |
+
MISTRAL_API_KEY=os.getenv("MISTRAL_API_KEY")
|
9 |
+
model=ChatMistralAI(
|
10 |
+
model="mistral-large-latest",
|
11 |
+
api_key=MISTRAL_API_KEY
|
12 |
+
)
|
13 |
+
|
14 |
+
|
15 |
+
with gr.Blocks(css=".gradio-container {background: url('image.png') !important;}") as Iface:
|
16 |
+
gr.Markdown("<h1>Search Shayari</h1>")
|
17 |
+
with gr.Row():
|
18 |
+
with gr.Column(scale=1):
|
19 |
+
query = gr.Textbox(label="Which type of shayari do you want to find?")
|
20 |
+
submit_button = gr.Button("Search")
|
21 |
+
|
22 |
+
with gr.Column(scale=2):
|
23 |
+
chatbot = gr.Chatbot(label="Matching results", height=600)
|
24 |
+
submit_button.click(
|
25 |
+
fn=llm,
|
26 |
+
inputs=[query, chatbot],
|
27 |
+
outputs=[chatbot, chatbot]
|
28 |
+
|
29 |
+
)
|
30 |
+
Iface.launch(share=True)
|
requirements.txt
ADDED
@@ -0,0 +1,183 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
aiofiles==23.2.1
|
2 |
+
aiohappyeyeballs==2.4.6
|
3 |
+
aiohttp==3.10.11
|
4 |
+
aiosignal==1.3.2
|
5 |
+
annotated-types==0.7.0
|
6 |
+
anyio==4.8.0
|
7 |
+
argon2-cffi==23.1.0
|
8 |
+
argon2-cffi-bindings==21.2.0
|
9 |
+
arrow==1.3.0
|
10 |
+
asttokens==3.0.0
|
11 |
+
async-lru==2.0.4
|
12 |
+
attrs==25.1.0
|
13 |
+
babel==2.17.0
|
14 |
+
beautifulsoup4==4.13.3
|
15 |
+
bleach==6.2.0
|
16 |
+
certifi==2025.1.31
|
17 |
+
cffi==1.17.1
|
18 |
+
charset-normalizer==3.4.1
|
19 |
+
click==8.1.8
|
20 |
+
cohere==5.13.12
|
21 |
+
colorama==0.4.6
|
22 |
+
comm==0.2.2
|
23 |
+
dataclasses-json==0.6.7
|
24 |
+
debugpy==1.8.12
|
25 |
+
decorator==5.1.1
|
26 |
+
defusedxml==0.7.1
|
27 |
+
distro==1.9.0
|
28 |
+
executing==2.2.0
|
29 |
+
fastapi==0.115.8
|
30 |
+
fastavro==1.10.0
|
31 |
+
fastjsonschema==2.21.1
|
32 |
+
ffmpy==0.5.0
|
33 |
+
filelock==3.17.0
|
34 |
+
fqdn==1.5.1
|
35 |
+
frozenlist==1.5.0
|
36 |
+
fsspec==2025.2.0
|
37 |
+
gradio==5.15.0
|
38 |
+
gradio_client==1.7.0
|
39 |
+
greenlet==3.1.1
|
40 |
+
h11==0.14.0
|
41 |
+
httpcore==1.0.7
|
42 |
+
httpx==0.28.1
|
43 |
+
httpx-sse==0.4.0
|
44 |
+
huggingface-hub==0.28.1
|
45 |
+
idna==3.10
|
46 |
+
iniconfig==2.0.0
|
47 |
+
inquirerpy==0.3.4
|
48 |
+
ipykernel==6.29.5
|
49 |
+
ipython==8.32.0
|
50 |
+
ipywidgets==8.1.5
|
51 |
+
isoduration==20.11.0
|
52 |
+
jedi==0.19.2
|
53 |
+
Jinja2==3.1.5
|
54 |
+
jiter==0.8.2
|
55 |
+
json5==0.10.0
|
56 |
+
jsonpatch==1.33
|
57 |
+
jsonpointer==3.0.0
|
58 |
+
jsonschema==4.23.0
|
59 |
+
jsonschema-specifications==2024.10.1
|
60 |
+
jupyter-console==6.6.3
|
61 |
+
jupyter-events==0.12.0
|
62 |
+
jupyter-lsp==2.2.5
|
63 |
+
jupyter_client==8.6.3
|
64 |
+
jupyter_core==5.7.2
|
65 |
+
jupyter_server==2.15.0
|
66 |
+
jupyter_server_terminals==0.5.3
|
67 |
+
jupyterlab==4.3.5
|
68 |
+
jupyterlab_pygments==0.3.0
|
69 |
+
jupyterlab_server==2.27.3
|
70 |
+
jupyterlab_widgets==3.0.13
|
71 |
+
langchain==0.3.18
|
72 |
+
langchain-cohere==0.4.2
|
73 |
+
langchain-community==0.3.17
|
74 |
+
langchain-core==0.3.34
|
75 |
+
langchain-mistralai==0.2.6
|
76 |
+
langchain-openai==0.3.4
|
77 |
+
langchain-pinecone==0.2.2
|
78 |
+
langchain-tests==0.3.11
|
79 |
+
langchain-text-splitters==0.3.6
|
80 |
+
langchain-together==0.3.0
|
81 |
+
langgraph==0.2.70
|
82 |
+
langgraph-checkpoint==2.0.10
|
83 |
+
langgraph-sdk==0.1.51
|
84 |
+
langsmith==0.3.7
|
85 |
+
markdown-it-py==3.0.0
|
86 |
+
MarkupSafe==2.1.5
|
87 |
+
marshmallow==3.26.1
|
88 |
+
matplotlib-inline==0.1.7
|
89 |
+
mdurl==0.1.2
|
90 |
+
mistune==3.1.1
|
91 |
+
msgpack==1.1.0
|
92 |
+
multidict==6.1.0
|
93 |
+
mypy-extensions==1.0.0
|
94 |
+
nbclient==0.10.2
|
95 |
+
nbconvert==7.16.6
|
96 |
+
nbformat==5.10.4
|
97 |
+
nest-asyncio==1.6.0
|
98 |
+
notebook_shim==0.2.4
|
99 |
+
numpy==1.26.4
|
100 |
+
openai==1.61.1
|
101 |
+
orjson==3.10.15
|
102 |
+
overrides==7.7.0
|
103 |
+
packaging==24.2
|
104 |
+
pandas==2.2.3
|
105 |
+
pandocfilters==1.5.1
|
106 |
+
parso==0.8.4
|
107 |
+
pfzy==0.3.4
|
108 |
+
pillow==11.1.0
|
109 |
+
pinecone==5.4.2
|
110 |
+
pinecone-plugin-inference==3.1.0
|
111 |
+
pinecone-plugin-interface==0.0.7
|
112 |
+
platformdirs==4.3.6
|
113 |
+
pluggy==1.5.0
|
114 |
+
prometheus_client==0.21.1
|
115 |
+
prompt_toolkit==3.0.50
|
116 |
+
propcache==0.2.1
|
117 |
+
psutil==6.1.1
|
118 |
+
pure_eval==0.2.3
|
119 |
+
pycparser==2.22
|
120 |
+
pydantic==2.10.6
|
121 |
+
pydantic-settings==2.7.1
|
122 |
+
pydantic_core==2.27.2
|
123 |
+
pydub==0.25.1
|
124 |
+
Pygments==2.19.1
|
125 |
+
pytest==8.3.4
|
126 |
+
pytest-asyncio==0.25.3
|
127 |
+
pytest-socket==0.7.0
|
128 |
+
python-dateutil==2.9.0.post0
|
129 |
+
python-dotenv==1.0.1
|
130 |
+
python-json-logger==3.2.1
|
131 |
+
python-multipart==0.0.20
|
132 |
+
pytz==2025.1
|
133 |
+
pywin32==308
|
134 |
+
pywinpty==2.0.15
|
135 |
+
PyYAML==6.0.2
|
136 |
+
pyzmq==26.2.1
|
137 |
+
referencing==0.36.2
|
138 |
+
regex==2024.11.6
|
139 |
+
requests==2.32.3
|
140 |
+
requests-toolbelt==1.0.0
|
141 |
+
rfc3339-validator==0.1.4
|
142 |
+
rfc3986-validator==0.1.1
|
143 |
+
rich==13.9.4
|
144 |
+
rpds-py==0.22.3
|
145 |
+
ruff==0.9.5
|
146 |
+
safehttpx==0.1.6
|
147 |
+
semantic-version==2.10.0
|
148 |
+
Send2Trash==1.8.3
|
149 |
+
shellingham==1.5.4
|
150 |
+
six==1.17.0
|
151 |
+
sniffio==1.3.1
|
152 |
+
soupsieve==2.6
|
153 |
+
SQLAlchemy==2.0.38
|
154 |
+
stack-data==0.6.3
|
155 |
+
starlette==0.45.3
|
156 |
+
syrupy==4.8.1
|
157 |
+
tenacity==9.0.0
|
158 |
+
terminado==0.18.1
|
159 |
+
tiktoken==0.8.0
|
160 |
+
tinycss2==1.4.0
|
161 |
+
tokenizers==0.21.0
|
162 |
+
tomlkit==0.13.2
|
163 |
+
tornado==6.4.2
|
164 |
+
tqdm==4.67.1
|
165 |
+
traitlets==5.14.3
|
166 |
+
typer==0.15.1
|
167 |
+
types-python-dateutil==2.9.0.20241206
|
168 |
+
types-PyYAML==6.0.12.20241230
|
169 |
+
types-requests==2.32.0.20241016
|
170 |
+
typing-inspect==0.9.0
|
171 |
+
typing_extensions==4.12.2
|
172 |
+
tzdata==2025.1
|
173 |
+
uri-template==1.3.0
|
174 |
+
urllib3==2.3.0
|
175 |
+
uvicorn==0.34.0
|
176 |
+
wcwidth==0.2.13
|
177 |
+
webcolors==24.11.1
|
178 |
+
webencodings==0.5.1
|
179 |
+
websocket-client==1.8.0
|
180 |
+
websockets==14.2
|
181 |
+
widgetsnbextension==4.0.13
|
182 |
+
yarl==1.18.3
|
183 |
+
zstandard==0.23.0
|
utilis.py
ADDED
@@ -0,0 +1,65 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import os
|
2 |
+
from dotenv import load_dotenv
|
3 |
+
from langchain_pinecone import PineconeVectorStore
|
4 |
+
from pinecone import Pinecone
|
5 |
+
from langchain_together import TogetherEmbeddings
|
6 |
+
from langchain_mistralai import ChatMistralAI
|
7 |
+
import cohere
|
8 |
+
from langchain.retrievers.contextual_compression import ContextualCompressionRetriever
|
9 |
+
from langchain_cohere import CohereRerank
|
10 |
+
|
11 |
+
|
12 |
+
load_dotenv()
|
13 |
+
PINECONE_API_KEY=os.getenv("PINECONE_API_KEY")
|
14 |
+
MISTRAL_API_KEY=os.getenv("MISTRAL_API_KEY")
|
15 |
+
TOGETHER_API=os.getenv("TOGETHER_API")
|
16 |
+
COHERE_API_KEY=os.getenv("COHERE_API_KEY")
|
17 |
+
|
18 |
+
co = cohere.Client(COHERE_API_KEY)
|
19 |
+
|
20 |
+
embeddings = TogetherEmbeddings(
|
21 |
+
model="togethercomputer/m2-bert-80M-8k-retrieval",
|
22 |
+
api_key=TOGETHER_API
|
23 |
+
)
|
24 |
+
|
25 |
+
model=ChatMistralAI(
|
26 |
+
model="mistral-large-latest",
|
27 |
+
api_key=MISTRAL_API_KEY
|
28 |
+
)
|
29 |
+
pc = Pinecone(api_key=PINECONE_API_KEY)
|
30 |
+
|
31 |
+
index_name = "together"
|
32 |
+
|
33 |
+
index = pc.Index(index_name)
|
34 |
+
|
35 |
+
vector_store = PineconeVectorStore(index=index, embedding=embeddings)
|
36 |
+
retriever = vector_store.as_retriever()
|
37 |
+
|
38 |
+
compressor = CohereRerank(model="rerank-v3.5")
|
39 |
+
compression_retriever = ContextualCompressionRetriever(
|
40 |
+
base_compressor=compressor, base_retriever=retriever
|
41 |
+
)
|
42 |
+
|
43 |
+
def retrieve(query):
|
44 |
+
compressed_docs = compression_retriever.invoke(query)
|
45 |
+
|
46 |
+
return compressed_docs[0]
|
47 |
+
|
48 |
+
def llm(query, history):
|
49 |
+
query=model.invoke(f"You are a helpful AI assistant. Translate the following query to urdu if the query is in urdu then return the same query. Query: {query}. Remember just translate the query and nothing else.").content
|
50 |
+
results=retrieve(query)
|
51 |
+
prompt=f""""
|
52 |
+
You are a helpful AI Assistant and your job is to help user find relevant shayari.
|
53 |
+
The query is {query}
|
54 |
+
The matching result after semantic search is {results.page_content}. You may exclude interpretation given in the results. Just respond from the context.
|
55 |
+
"""
|
56 |
+
|
57 |
+
try:
|
58 |
+
response=model.invoke(prompt).content
|
59 |
+
history = history or []
|
60 |
+
history.append(("You", query))
|
61 |
+
history.append(("Assistant", results.page_content))
|
62 |
+
return history, history
|
63 |
+
except Exception as e:
|
64 |
+
return history, history + [("Bot", f"Error in API request: {str(e)}")]
|
65 |
+
|