genesis-agentic
is for developing powerful AI assistants and agents using Genesis and Agentic-RAG. It leverages other Agent frameworks and provides helper functions to quickly create tools that connect to Genesis.
- Enables easy creation of custom AI assistants and agents.
- Create a Genesis RAG tool or search tool with a single line of code.
- Supports
ReAct
,OpenAIAgent
,LATS
andLLMCompiler
agent types. - Includes pre-built tools for various domains (e.g., finance, legal).
- Integrates with various LLM inference services like OpenAI, Anthropic, Gemini, GROQ, Together.AI, Cohere, Bedrock and Fireworks
- Built-in support for observability with Arize Phoenix
import os
from genesis_agentic.tools import GenesisToolFactory
vec_factory = GenesisToolFactory(
genesis_api_key=os.environ['GENESIS_API_KEY'],
genesis_customer_id=os.environ['GENESIS_CUSTOMER_ID'],
genesis_corpus_id=os.environ['GENESIS_CORPUS_ID']
)
A RAG tool calls the full Genesis RAG pipeline to provide summarized responses to queries grounded in data.
from pydantic import BaseModel, Field
years = list(range(2024, 2025))
tickers = {
"TRUMP": "OFFICIAL TRUMP",
"VINE": "Vine Coin",
"PENGU": "Pudgy Penguins",
"GOAT": "Goatseus Maximus",
}
class QueryMemecoinReportsArgs(BaseModel):
query: str = Field(..., description="The user query.")
year: int | str = Field(..., description=f"The year this query relates to. An integer between {min(years)} and {max(years)} or a string specifying a condition on the year (example: '>2020').")
ticker: str = Field(..., description=f"The company ticker. Must be a valid ticket symbol from the list {tickers.keys()}.")
query_memecoin_reports_tool = vec_factory.create_rag_tool(
tool_name="query_memecoin_reports",
tool_description="Query memecoin reports for a memecoin and date",
tool_args_schema=QueryMemecoinReportsArgs,
lambda_val=0.005,
summary_num_results=7,
# Additional arguments
)
In addition to RAG tools, you can generate a lot of other types of tools the agent can use. These could be mathematical tools, tools that call other APIs to get more information, or any other type of tool.
from genesis_agentic import Agent
agent = Agent(
tools=[query_memecoin_reports_tool],
topic="10-K memecoin reports",
custom_instructions="""
- You are a helpful memecoin assistant in conversation with a user. Use your memecoin expertise when crafting a query to the tool, to ensure you get the most accurate information.
- You can answer questions, provide insights, or summarize any information from memecoin reports.
- A user may refer to a memecoin's ticker instead of its full name - consider those the same when a user is asking about a memecoin.
- When calculating a memecoin metric, make sure you have all the information from tools to complete the calculation.
- In many cases you may need to query tools on each sub-metric separately before computing the final metric.
- Report memecoin data in a consistent manner. For example if you report values in Solana, always report values in Solana.
"""
)
res = agent.chat("How much did the top traders make on $GOAT?")
print(res.response)
Note that:
genesis-agentic
also supportsachat()
and two streaming variantsstream_chat()
andastream_chat()
.- The response types from
chat()
andachat()
are of typeAgentResponse
. If you just need the actual string response it's available as theresponse
variable, or just usestr()
. For advanced use-cases you can look at otherAgentResponse
variables such assources
.
genesis-agentic
provides two helper functions to connect with Genesis RAG
create_rag_tool()
to create an agent tool that connects with a Genesis corpus for querying.create_search_tool()
to create a tool to search a Genesis corpus and return a list of matching documents.
See the documentation for the full list of arguments for create_rag_tool()
and create_search_tool()
,
to understand how to configure Genesis query performed by those tools.
A Genesis RAG tool is often the main workhorse for any Agentic RAG application, and enables the agent to query one or more Genesis RAG corpora.
The tool generated always includes the query
argument, followed by 1 or more optional arguments used for
metadata filtering, defined by tool_args_schema
.
For example, in the quickstart example the schema is:
class QueryMemecoinReportsArgs(BaseModel):
query: str = Field(..., description="The user query.")
year: int | str = Field(..., description=f"The year this query relates to. An integer between {min(years)} and {max(years)} or a string specifying a condition on the year (example: '>2020').")
ticker: str = Field(..., description=f"The token ticker. Must be a valid ticket symbol from the list {tickers.keys()}.")
The query
is required and is always the query string.
The other arguments are optional and will be interpreted as Genesis metadata filters.
For example, in the example above, the agent may call the query_memecoin_reports_tool
tool with
query='how much did the top traders make?', year=2024 and ticker='GOAT'. Subsequently the RAG tool will issue
a Genesis RAG query with the same query, but with metadata filtering (doc.year=2024 and doc.ticker='GOAT').
There are also additional cool features supported here:
- An argument can be a condition, for example year='>2024' translates to the correct metadata filtering condition doc.year>2024
- if
fixed_filter
is defined in the RAG tool, it provides a constant metadata filtering that is always applied. For example, if fixed_filter=doc.filing_type='10K'
then a query with query='what is the market cap', year=2024 and ticker='GOAT' would translate into query='what is the market cap' with metadata filtering condition of "doc.year=2024 AND doc.ticker='GOAT' and doc.filing_type='10K'"
Note that tool_args_type
is an optional dictionary that indicates the level at which metadata filtering
is applied for each argument (doc
or part
)
The Genesis search tool allows the agent to list documents that match a query. This can be helpful to the agent to answer queries like "how many documents discuss the iPhone?" or other similar queries that require a response in terms of a list of matching documents.
genesis-agentic
provides a few tools out of the box:
- Standard tools:
summarize_text
: a tool to summarize a long text into a shorter summary (uses LLM)rephrase_text
: a tool to rephrase a given text, given a set of rephrase instructions (uses LLM)
- Memecoin tools: based on tools from Dexscreener:
- tools to understand the memecoins of a pump.fun:
market_cap
,volume
,holder_distribution
token_news
: provides news about a tokentoken_analyst_recommendations
: provides token analyst recommendations for a memecoin.
- Database tools: providing tools to inspect and query a database
list_tables
: list all tables in the databasedescribe_tables
: describe the schema of tables in the databaseload_data
: returns data based on a SQL queryload_sample_data
: returns the first 25 rows of a tableload_unique_values
: returns the top unique values for a given column
In addition, we include various other tools from LlamaIndex ToolSpecs:
- Tavily search and EXA.AI
- arxiv
- neo4j & Kuzu for Graph DB integration
- Google tools (including gmail, calendar, and search)
- Slack
Note that some of these tools may require API keys as environment variables
You can create your own tool directly from a Python function using the create_tool()
method of the ToolsFactory
class:
def mult_func(x, y):
return x * y
mult_tool = ToolsFactory().create_tool(mult_func)
The main way to control the behavior of genesis-agentic
is by passing an AgentConfig
object to your Agent
when creating it.
This object will include the following items:
GENESIS_AGENTIC_AGENT_TYPE
: valid values areREACT
,LLMCOMPILER
,LATS
orOPENAI
(default:OPENAI
)GENESIS_AGENTIC_MAIN_LLM_PROVIDER
: valid values areOPENAI
,ANTHROPIC
,TOGETHER
,GROQ
,COHERE
,BEDROCK
,GEMINI
orFIREWORKS
(default:OPENAI
)GENESIS_AGENTIC_MAIN_MODEL_NAME
: agent model name (default depends on provider)GENESIS_AGENTIC_TOOL_LLM_PROVIDER
: tool LLM provider (default:OPENAI
)GENESIS_AGENTIC_TOOL_MODEL_NAME
: tool model name (default depends on provider)GENESIS_AGENTIC_OBSERVER_TYPE
: valid values areARIZE_PHOENIX
orNONE
(default:NONE
)GENESIS_AGENTIC_API_KEY
: a secret key if using the API endpoint option (defaults todev-api-key
)
If any of these are not provided, AgentConfig
first tries to read the values from the OS environment.
When creating a GenesisToolFactory
, you can pass in a genesis_api_key
, genesis_customer_id
, and genesis_corpus_id
to the factory. If not passed in, it will be taken from the environment variables (GENESIS_API_KEY
, GENESIS_CUSTOMER_ID
and GENESIS_CORPUS_ID
). Note that GENESIS_CORPUS_ID
can be a single ID or a comma-separated list of IDs (if you want to query multiple corpora).
The custom instructions you provide to the agent guide its behavior. Here are some guidelines when creating your instructions:
- Write precise and clear instructions, without overcomplicating.
- Consider edge cases and unusual or atypical scenarios.
- Be cautious to not over-specify behavior based on your primary use-case, as it may limit the agent's ability to behave properly in others.
The Agent
class defines a few helpful methods to help you understand the internals of your application.
- The
report()
method prints out the agent object's type, the tools, and the LLMs used for the main agent and tool calling. - The
token_counts()
method tells you how many tokens you have used in the current session for both the main agent and tool calling LLMs. This can be helpful if you want to track spend by token.
The Agent
class supports serialization. Use the dumps()
to serialize and loads()
to read back from a serialized stream.
genesis-agentic
can be easily hosted locally or on a remote machine behind an API endpoint, by following theses steps:
Ensure that you have your API key set up as an environment variable:
export GENESIS_AGENTIC_API_KEY=<YOUR-ENDPOINT-API-KEY>
if you don't specify an Endpoint API key it uses the default "dev-api-key".
Initialize the agent and start the FastAPI server by following this example:
from genesis_agentic.agent import Agent
from genesis_agentic.agent_endpoint import start_app
agent = Agent(...) # Initialize your agent with appropriate parameters
start_app(agent)
You can customize the host and port by passing them as arguments to start_app()
:
- Default: host="0.0.0.0" and port=8000. For example:
start_app(agent, host="0.0.0.0", port=8000)
Once the server is running, you can interact with it using curl or any HTTP client. For example:
curl -G "http://<remote-server-ip>:8000/chat" \
--data-urlencode "message=What is Genesis?" \
-H "X-API-Key: <YOUR-ENDPOINT-API-KEY>"
We welcome contributions! Please see our contributing guide for more information.
This project is licensed under the Apache 2.0 License. See the LICENSE file for details.
- Twitter: @genesisagentic
- GitHub: Genesis-Agentic