Update app.py
Browse files
app.py
CHANGED
|
@@ -469,6 +469,12 @@ import numpy as np
|
|
| 469 |
from langchain.schema import SystemMessage, HumanMessage
|
| 470 |
from langchain.chat_models import ChatOpenAI
|
| 471 |
from sklearn.metrics.pairwise import cosine_similarity
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 472 |
|
| 473 |
EMBEDDING_FILE = "embeddings.json"
|
| 474 |
EMBEDDING_MODEL = "intfloat/multilingual-e5-large-instruct"
|
|
@@ -521,22 +527,23 @@ if query:
|
|
| 521 |
try:
|
| 522 |
query_embedding = get_query_embedding_together(query)
|
| 523 |
data = load_embeddings(EMBEDDING_FILE)
|
| 524 |
-
top_chunks = find_most_similar_chunks(query_embedding, data, top_n=
|
|
|
|
| 525 |
|
| 526 |
context = "\n".join(top_chunks)
|
| 527 |
prompt = f"""
|
| 528 |
-
|
| 529 |
-
|
| 530 |
-
|
| 531 |
سؤال:
|
| 532 |
{query}
|
| 533 |
-
|
| 534 |
محتوا:
|
| 535 |
{context}
|
| 536 |
-
|
| 537 |
-
|
| 538 |
"""
|
| 539 |
|
|
|
|
| 540 |
response = llm([
|
| 541 |
SystemMessage(
|
| 542 |
content="تو یک دستیار دقیق هستی که فقط با اطلاعات موجود در متن پاسخ میدهی. اگر اطلاعات نبود، آن را اعلام میکنی و بعد از دانش خودت استفاده میکنی."
|
|
|
|
| 469 |
from langchain.schema import SystemMessage, HumanMessage
|
| 470 |
from langchain.chat_models import ChatOpenAI
|
| 471 |
from sklearn.metrics.pairwise import cosine_similarity
|
| 472 |
+
llm = ChatOpenAI(
|
| 473 |
+
base_url="https://api.together.xyz/v1",
|
| 474 |
+
api_key='333ac33f5be91819cb7ade101134d73f5e63d299a964ae290850eeac5d82a8d5',
|
| 475 |
+
model="Qwen/Qwen2-VL-72B-Instruct",
|
| 476 |
+
|
| 477 |
+
)
|
| 478 |
|
| 479 |
EMBEDDING_FILE = "embeddings.json"
|
| 480 |
EMBEDDING_MODEL = "intfloat/multilingual-e5-large-instruct"
|
|
|
|
| 527 |
try:
|
| 528 |
query_embedding = get_query_embedding_together(query)
|
| 529 |
data = load_embeddings(EMBEDDING_FILE)
|
| 530 |
+
top_chunks = find_most_similar_chunks(query_embedding, data, top_n=10)
|
| 531 |
+
st.write("Top Chunks:", top_chunks)
|
| 532 |
|
| 533 |
context = "\n".join(top_chunks)
|
| 534 |
prompt = f"""
|
| 535 |
+
لطفاً سعی کن با استفاده از اطلاعات زیر به سؤال پاسخ بده. اگر اطلاعات کافی نبود، ابتدا بگو اطلاعات ناقص است و بعد اگر لازم بود از دانش عمومیات استفاده کن.
|
| 536 |
+
|
|
|
|
| 537 |
سؤال:
|
| 538 |
{query}
|
| 539 |
+
|
| 540 |
محتوا:
|
| 541 |
{context}
|
| 542 |
+
|
| 543 |
+
پاسخ:
|
| 544 |
"""
|
| 545 |
|
| 546 |
+
|
| 547 |
response = llm([
|
| 548 |
SystemMessage(
|
| 549 |
content="تو یک دستیار دقیق هستی که فقط با اطلاعات موجود در متن پاسخ میدهی. اگر اطلاعات نبود، آن را اعلام میکنی و بعد از دانش خودت استفاده میکنی."
|