File size: 2,612 Bytes
c595ba7
acd6dc5
 
 
 
 
af7fd9b
acd6dc5
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
9498171
acd6dc5
9498171
9c46c22
c595ba7
9498171
 
 
 
 
 
 
 
 
 
1665886
9498171
 
 
 
 
 
 
 
 
 
 
 
 
 
c595ba7
 
 
9498171
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
import gradio as gr
from paperqa import Docs, SentenceTransformerEmbeddingModel
from langchain_anthropic import ChatAnthropic

MODEL_NAME = "claude-3-5-sonnet-20240620"
class MyChatAnthropic(ChatAnthropic):
    model_name: str = MODEL_NAME
llm = MyChatAnthropic(
    model=MODEL_NAME,
    temperature=0.2,
    max_tokens=4096,)

class MyEmb(SentenceTransformerEmbeddingModel):
    async def aembed_documents(self, texts):
        return await self.embed_documents(None, texts)

emb = MyEmb(model_name="mixedbread-ai/mxbai-embed-large-v1")
docs = Docs(llm="langchain",
            embedding="langchain",
            embedding_client=emb,
            client=llm)
docs.max_concurrent = 1
docs.add('knowledge_extraction.csv', disable_check=True)
docs.add('SSA - POMS_ SI 00502.100 - Basic SSI Alien Eligibility Requirements - 11_30_2023.pdf', disable_check=True)

def respond(message, *args, **kwargs):
    return docs.query(message).answer


examples = [["A qualified alien entered the United States on August 15, 1996. Tell me about their eligibility for Supplemental Security Income (SSI)."],
            ["""Question: Which of the following statements most accurately reflects the eligibility requirements for Supplemental Security Income (SSI) as outlined in the Alien Eligibility Under Welfare Reform?

A) To be eligible for SSI, an individual must be a U.S. citizen, with no exceptions for non-citizens.

B) SSI eligibility is extended to all individuals residing in the United States, regardless of their citizenship or immigration status.

C) To be eligible for SSI, an individual must be either a U.S. citizen or a qualified alien.

D) SSI eligibility is determined solely based on financial need, without consideration of citizenship or immigration status."""],
           ["""Explanate the Correct Answer: C
Question: Which of the following statements most accurately reflects the eligibility requirements for Supplemental Security Income (SSI) as outlined in the Alien Eligibility Under Welfare Reform?

A) To be eligible for SSI, an individual must be a U.S. citizen, with no exceptions for non-citizens.

B) SSI eligibility is extended to all individuals residing in the United States, regardless of their citizenship or immigration status.

C) To be eligible for SSI, an individual must be either a U.S. citizen or a qualified alien.

D) SSI eligibility is determined solely based on financial need, without consideration of citizenship or immigration status."""]]

demo = gr.ChatInterface(
    respond,
    examples=examples)


if __name__ == "__main__":
    demo.launch(auth=("tester", "8888p4ss"))