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="تو یک دستیار دقیق هستی که فقط با اطلاعات موجود در متن پاسخ میدهی. اگر اطلاعات نبود، آن را اعلام میکنی و بعد از دانش خودت استفاده میکنی."
|