b6b4cd0a4c
### Overview Major restructuring from monolithic `arcade-ai` package to modular library architecture with standardized uv-based dependency management.  ### New Package Structure - **`arcade-tdk`** - Lightweight toolkit development kit (core decorators, auth) - **`arcade-core`** - Core execution engine and catalog functionality - **`arcade-serve`** - FastAPI/MCP server components - **`arcade-ai`** - Meta package that includes CLI functionality. Optionally include evals via the `evals` extra. Optionally include all packages via the `all` extra. ### Key Benefits - **Lighter Dependencies**: Toolkits now depend only on `arcade-tdk` (~2 deps) vs full `arcade-ai` (~30+ deps) - **Faster Builds**: uv provides 10-100x faster dependency resolution and installation - **Better Modularity**: Clear separation of concerns, consumers import only what they need - **Standard Tooling**: Eliminates custom poetry scripts, uses standard Python packaging ### Migration Impact - All 20 toolkits converted from poetry → uv with `arcade-tdk` dependencies plus `arcade-ai[evals]` and `arcade-serve` dev dependencies. When developing locally, devs should install toolkits via `make install-local`. - Modern Python 3.10+ type hints throughout - Standardized build system with hatchling backend - Enhanced Makefile with robust toolkit management commands - Removed `arcade dev` CLI command - Reduce the number of files created by `arcade new` and add an option to not generate a tests and evals folder. This foundation enables faster development cycles and cleaner dependency chains for the growing toolkit ecosystem. ### Todo After this PR is merged - [ ] Post-merge workflow(s) (release & publish containers, etc) - [ ] Release order plan. @EricGustin suggests releasing in the following order: 1. `arcade-core` version 0.1.0 2. `arcade-serve` version 0.1.0 and `arcade-tdk` version 0.1.0 3. `arcade-ai` version 2.0.0 4. Patch release for all toolkits (all changes in toolkits are internal refactors) - [ ] [Update docs](https://github.com/ArcadeAI/docs/pull/318) --------- Co-authored-by: Eric Gustin <eric@arcade.dev> Co-authored-by: Eric Gustin <34000337+EricGustin@users.noreply.github.com>
86 lines
2.7 KiB
Python
86 lines
2.7 KiB
Python
from typing import Annotated, Any
|
|
|
|
from arcade_tdk import ToolContext, tool
|
|
|
|
from arcade_search.enums import GoogleFinanceWindow
|
|
from arcade_search.utils import call_serpapi, prepare_params
|
|
|
|
|
|
@tool(requires_secrets=["SERP_API_KEY"])
|
|
async def get_stock_summary(
|
|
context: ToolContext,
|
|
ticker_symbol: Annotated[
|
|
str,
|
|
"The stock ticker to get summary for. For example, 'GOOG' is the ticker symbol for Google",
|
|
],
|
|
exchange_identifier: Annotated[
|
|
str,
|
|
"The exchange identifier. This part indicates the market where the "
|
|
"stock is traded. For example, 'NASDAQ', 'NYSE', 'TSE', 'LSE', etc.",
|
|
],
|
|
) -> Annotated[dict[str, Any], "Summary of the stock's recent performance"]:
|
|
"""Retrieve the summary information for a given stock ticker using the Google Finance API.
|
|
|
|
Gets the stock's current price as well as price movement from the most recent trading day.
|
|
"""
|
|
# Prepare the request
|
|
query = (
|
|
f"{ticker_symbol.upper()}:{exchange_identifier.upper()}"
|
|
if exchange_identifier
|
|
else ticker_symbol.upper()
|
|
)
|
|
params = prepare_params("google_finance", q=query)
|
|
|
|
# Execute the request
|
|
results = call_serpapi(context, params)
|
|
|
|
# Parse the results
|
|
summary: dict = results.get("summary", {})
|
|
|
|
return summary
|
|
|
|
|
|
@tool(requires_secrets=["SERP_API_KEY"])
|
|
async def get_stock_historical_data(
|
|
context: ToolContext,
|
|
ticker_symbol: Annotated[
|
|
str,
|
|
"The stock ticker to get summary for. For example, 'GOOG' is the ticker symbol for Google",
|
|
],
|
|
exchange_identifier: Annotated[
|
|
str,
|
|
"The exchange identifier. This part indicates the market where the "
|
|
"stock is traded. For example, 'NASDAQ', 'NYSE', 'TSE', 'LSE', etc.",
|
|
],
|
|
window: Annotated[
|
|
GoogleFinanceWindow, "Time window for the graph data. Defaults to 1 month"
|
|
] = GoogleFinanceWindow.ONE_MONTH,
|
|
) -> Annotated[
|
|
dict[str, Any],
|
|
"A stock's price and volume data at a specific time interval over a specified time window",
|
|
]:
|
|
"""Fetch historical stock price data over a specified time window
|
|
|
|
Returns a stock's price and volume data over a specified time window
|
|
"""
|
|
# Prepare the request
|
|
query = (
|
|
f"{ticker_symbol.upper()}:{exchange_identifier.upper()}"
|
|
if exchange_identifier
|
|
else ticker_symbol.upper()
|
|
)
|
|
params = prepare_params("google_finance", q=query, window=window.value)
|
|
|
|
# Execute the request
|
|
results = call_serpapi(context, params)
|
|
|
|
# Parse the results
|
|
data = {
|
|
"summary": results.get("summary", {}),
|
|
"graph": results.get("graph", []),
|
|
}
|
|
key_events = results.get("key_events")
|
|
if key_events:
|
|
data["key_events"] = key_events
|
|
|
|
return data
|