Search & RAG

1
search_response = client.search("What was Uber's profit in 2020?")

1
# returns only chunks from documents with title `document_title`
2
filtered_search_response = client.search(
3
    "What was Uber's profit in 2020?",
4
    vector_search_settings={
5
        # restrict results to the Uber document
6
        "filters": {"title": {"$eq": "uber_2021.pdf"}},
7
        "search_limit": 100
8
    }
9
)

1
hybrid_search_response = client.search(
2
    "What was Uber's profit in 2020?",
3
    vector_search_settings={
4
        "use_hybrid_search": True,
5
        "search_limit": 20,
6
        "hybrid_search_settings": {
7
            "full_text_weight": 1.0,
8
            "semantic_weight": 10.0,
9
            "full_text_limit": 200,
10
            "rrf_k": 25,
11
        },
12
    }
13
)

1
kg_search_response = client.search(
2
    "What is airbnb",
3
    vector_search_settings={"use_vector_search": False},
4
    kg_search_settings={
5
      "use_kg_search": True,
6
      "kg_search_type": "local",
7
      "kg_search_level": "0",
8
      "generation_config": {
9
          "model": "openai/gpt-4o-mini",
10
          "temperature": 0.7,
11
      },
12
      "local_search_limits": {
13
          "__Entity__": 20,
14
          "__Relationship__": 20,
15
          "__Community__": 20,
16
      },
17
      "max_community_description_length": 65536,
18
      "max_llm_queries_for_global_search": 250
19
    }
20
)

1
rag_response = client.rag("What was Uber's profit in 2020?")

1
hybrid_rag_response = client.rag(
2
    "Who is Jon Snow?",
3
    vector_search_settings={"use_hybrid_search": True}
4
)

1
anthropic_rag_response = client.rag(
2
    "What is R2R?",
3
    rag_generation_config={"model":"anthropic/claude-3-opus-20240229"}
4
)

1
stream_response = client.rag(
2
    "Who was Aristotle?",
3
    rag_generation_config={"stream": True}
4
)
5
for chunk in stream_response:
6
    print(chunk, end='', flush=True)

1
from r2r import R2RClient
2
3
client = R2RClient()
4
5
# Using HyDE
6
hyde_response = client.rag(
7
    "What are the main themes in Shakespeare's plays?",
8
    vector_search_settings={
9
        "search_strategy": "hyde",
10
        "search_limit": 10
11
    }
12
)
13
14
# Using RAG-Fusion
15
rag_fusion_response = client.rag(
16
    "Explain the theory of relativity",
17
    vector_search_settings={
18
        "search_strategy": "rag_fusion",
19
        "search_limit": 20
20
    }
21
)

1
custom_rag_response = client.rag(
2
    "Who was Aristotle?",
3
    vector_search_settings={
4
        "use_hybrid_search": True,
5
        "search_limit": 20,
6
        "hybrid_search_settings": {
7
            "full_text_weight": 1.0,
8
            "semantic_weight": 10.0,
9
            "full_text_limit": 200,
10
            "rrf_k": 25,
11
        },
12
    },
13
    kg_search_settings={
14
        "use_kg_search": True,
15
        "kg_search_type": "local",
16
    },
17
    rag_generation_config={
18
        "model": "anthropic/claude-3-haiku-20240307",
19
        "temperature": 0.7,
20
        "stream": True
21
    },
22
    task_prompt_override="Only answer the question if the context is SUPER relevant!!\n\nQuery:\n{query}\n\nContext:\n{context}"
23
)

1
messages = [
2
    {"role": "user", "content": "What was Aristotle's main contribution to philosophy?"},
3
    {"role": "assistant", "content": "Aristotle made numerous significant contributions to philosophy, but one of his main contributions was in the field of logic and reasoning. He developed a system of formal logic, which is considered the first comprehensive system of its kind in Western philosophy. This system, often referred to as Aristotelian logic or term logic, provided a framework for deductive reasoning and laid the groundwork for scientific thinking."},
4
    {"role": "user", "content": "Can you elaborate on how this influenced later thinkers?"}
5
]
6
7
rag_agent_response = client.agent(
8
    messages,
9
    vector_search_settings={"use_hybrid_search":True},
10
)

1
messages = [
2
    {"role": "user", "content": "What was Aristotle's main contribution to philosophy?"},
3
    {"role": "assistant", "content": "Aristotle made numerous significant contributions to philosophy, but one of his main contributions was in the field of logic and reasoning. He developed a system of formal logic, which is considered the first comprehensive system of its kind in Western philosophy. This system, often referred to as Aristotelian logic or term logic, provided a framework for deductive reasoning and laid the groundwork for scientific thinking."},
4
    {"role": "user", "content": "Can you elaborate on how this influenced later thinkers?"}
5
]
6
7
rag_agent_response = client.agent(
8
    messages,
9
    vector_search_settings={"use_hybrid_search":True},
10
    rag_generation_config={"stream":True}
11
)