Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
https://github.com/SciPhi-AI/agent-search
AgentSearch is a framework for powering search agents and enabling customizable local search.
https://github.com/SciPhi-AI/agent-search
artificial-intelligence information-retrieval llms rag retrieval-augmented-generation search search-engine
Last synced: 14 days ago
JSON representation
AgentSearch is a framework for powering search agents and enabling customizable local search.
- Host: GitHub
- URL: https://github.com/SciPhi-AI/agent-search
- Owner: SciPhi-AI
- License: apache-2.0
- Created: 2023-12-11T17:41:03.000Z (11 months ago)
- Default Branch: main
- Last Pushed: 2024-04-22T20:54:17.000Z (7 months ago)
- Last Synced: 2024-09-22T20:16:23.954Z (about 2 months ago)
- Topics: artificial-intelligence, information-retrieval, llms, rag, retrieval-augmented-generation, search, search-engine
- Language: Python
- Homepage: https://agent-search.readthedocs.io/en/latest/
- Size: 261 KB
- Stars: 433
- Watchers: 3
- Forks: 45
- Open Issues: 4
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# AgentSearch: A framework for powering search agents and enabling customizable local search.
AgentSearch is a framework for powering search agents by seamlessly integrating LLM technologies from various providers with different search engines. This integration enables search agents to perform a wide range of functions through Retrieval-Augmented Generation (RAG), including summarizing search results, generating new queries, and retrieving detailed downstream results.
## Features of AgentSearch
- **Search Agent Integration**: Effortlessly build a search agent by connecting any search-specialized LLM, such as [Sensei-7B](https://huggingface.co/SciPhi/Sensei-7B-V1), with a supported search engine.
- **Customizable Search**: Utilize the [AgentSearch dataset](https://huggingface.co/datasets/SciPhi/AgentSearch-V1) in conjunction with this framework to deploy a customizable local search engine.
- **API Endpoint Integration**: Seamlessly integrate with a variety of hosted provider APIs for diverse search solutions, offering ease of use and flexibility, including Bing, SERP API, and AgentSearch. Additionally, support is provided for LLMs from SciPhi, HuggingFace, OpenAI, Anthropic, and more.
## Quickstart Guide
### Installation
```bash
pip install agent-search
```
### Configuration
Get your free API key from [SciPhi](https://www.sciphi.ai/signup) and set it in your environment:
```bash
export SCIPHI_API_KEY=$MY_SCIPHI_API_KEY
```
### Usage
Call a pre-configured search agent endpoint:
```python
# Requires SCIPHI_API_KEY in the environment
from agent_search import SciPhi
client = SciPhi()
# Search, then summarize result and generate related queries
agent_summary = client.get_search_rag_response(query='latest news', search_provider='bing', llm_model='SciPhi/Sensei-7B-V1')
print(agent_summary)
# { 'response': '...', 'other_queries': '...', 'search_results': '...' }
```
Standalone searches and from the AgentSearch search engine are supported:
```python
# Requires SCIPHI_API_KEY in the environment
from agent_search import SciPhi
client = SciPhi()
# Perform a search
search_response = client.search(query='Quantum Field Theory', search_provider='agent-search')
print(search_response)
# [{ 'score': '.89', 'url': 'https://...', 'metadata': {...} }]
```
Code your own custom search agent workflow:
```python
# Requires SCIPHI_API_KEY in the environment
from agent_search import SciPhi
client = SciPhi()
# Specify instructions for the task
instruction = "Your task is to perform retrieval augmented generation (RAG) over the given query and search results. Return your answer in a json format that includes a summary of the search results and a list of related queries."
query = "What is Fermat's Last Theorem?"
# construct search context
search_response = client.search(query=query, search_provider='agent-search')
search_context = "\n\n".join(
f"{idx + 1}. Title: {item['title']}\nURL: {item['url']}\nText: {item['text']}"
for idx, item in enumerate(search_response)
).encode('utf-8')
# Prefix to enforce a JSON response
json_response_prefix = '{"summary":'
# Prepare a prompt
formatted_prompt = f"### Instruction:{instruction}\n\nQuery:\n{query}\n\nSearch Results:\n${search_context}\n\nQuery:\n{query}\n### Response:\n{json_response_prefix}",
# Generate a completion with Sensei-7B-V1
completion = json_response_prefix + client.completion(formatted_prompt, llm_model_name="SciPhi/Sensei-7B-V1")
print(completion)
# {
# "summary": "\nFermat's Last Theorem is a mathematical proposition first prop ... ",
# "other_queries": ["The role of elliptic curves in the proof of Fermat's Last Theorem", ...]
# }
```
## Community & Support
- **Engage with Us:** Join our [Discord community](https://discord.gg/mN4kWbsgRu) for discussions and updates.
- **Feedback & Inquiries:** Contact us via email for personalized support.
### Additional Notes
- Execute commands from the root directory of the AgentSearch project.
- User Guide coming soon!