Spaces:
Sleeping
Sleeping
from langchain_openai import ChatOpenAI | |
from langchain.prompts.prompt import PromptTemplate | |
from typing import Tuple, List | |
from langchain.schema import format_document | |
import gradio as gr | |
from langchain.chat_models import ChatOpenAI | |
import os | |
from langchain_openai import ChatOpenAI | |
import os | |
DEFAULT_DOCUMENT_PROMPT = PromptTemplate.from_template(template="{page_content}") | |
def make_pairs(lst): | |
"""from a list of even lenght, make tupple pairs""" | |
return [(lst[i], lst[i + 1]) for i in range(0, len(lst), 2)] | |
def reset_textbox(): | |
return gr.update(value="") | |
def _combine_documents( | |
docs, document_prompt=DEFAULT_DOCUMENT_PROMPT, document_separator="\n\n" | |
): | |
doc_strings = [f"Document {i}: \n'''\n{format_document(doc, document_prompt)}\n'''" for i, doc in enumerate(docs, 1)] | |
return document_separator.join(doc_strings) | |
def _format_chat_history(chat_history: List[Tuple]) -> str: | |
buffer = "" | |
for dialogue_turn in chat_history: | |
human = "Human: " + dialogue_turn[0] | |
ai = "Assistant: " + dialogue_turn[1] | |
buffer += "\n" + "\n".join([human, ai]) | |
return buffer | |
def _format_chat_history(chat_history: List[Tuple]) -> str: | |
turn = 1 | |
buffer = [] | |
for dialogue in chat_history: | |
buffer.append(("Human: " if turn else "Assistant: ") + dialogue.content) | |
turn ^= 1 | |
return "\n".join(buffer) + "\n" | |
def get_llm(model="gpt-4o-mini",max_tokens=1024, temperature=0.0, streaming=True,timeout=30, **kwargs): | |
llm = ChatOpenAI( | |
model=model, | |
api_key=os.environ.get("OPENAI_API_KEY", None), | |
max_tokens = max_tokens, | |
streaming = streaming, | |
temperature=temperature, | |
timeout = timeout, | |
**kwargs, | |
) | |
return llm | |
def make_html_source(source,i): | |
meta = source.metadata | |
# content = source.page_content.split(":",1)[1].strip() | |
content = source.page_content.strip() | |
card = f""" | |
<div class="card" id="doc{i}"> | |
<div class="card-content"> | |
<h2>Document {i} - Meeting {meta["meeting_number"]} - title {meta['Title']} - Issues {meta['Issues']}</h2> | |
<p>{content}</p> | |
</div> | |
</div> | |
""" | |
return card | |