Feature cookie api #22

AlexM369 · 2025-01-31T18:58:38Z

This PR integrates three new API endpoints from cookie.fun into the repo, enabling retrieval of agent details via different identifiers. The changes include:
Get Agent by Twitter Username
• Retrieves agent details based on a given Twitter username at a specified interval.
• Example: "get agent metrics for twitter username 'griffaindotcom'"

Get Agent by Contract Address
• Retrieves agent details based on a supported token contract address.
• Also enables token symbol lookup using ChainConfig.get_token_info_or_none() from config.py.
• Example: "What are mindshare metrics for SEKOIA?" or "Mindshare change of 0xc0041ef357b183448b235a8ea73ce4e4ec8c265f"

Get Agents Paged
• Retrieves a paginated list of agents ordered by 7-day mindshare (descending).
• Example: "What are top 25 AI agents?"

catch up with main

gkoch78 · 2025-01-31T19:03:22Z

alphaswarm/services/cookiefun/cookiefun_client.py

+        except Exception:
+            logger.exception(f"Failed to find token address for {symbol}")
+            return None, None


it would be better to raise the exception here, rather than return (None, None) to differentiate between actually no result and error in execution

Suggested change

except Exception:

logger.exception(f"Failed to find token address for {symbol}")

return None, None

except Exception:

logger.exception(f"Failed to find token address for {symbol}")

raise

gkoch78 · 2025-01-31T19:06:55Z

alphaswarm/services/cookiefun/cookiefun_client.py

+
+@dataclass
+class Tweet:
+    tweetUrl: str


property name in python should be lower_snake_case. To enable proper mapping with API schema, use tweet_url: str = Field(alias="tweet_url")

made changes

gkoch78 · 2025-01-31T19:08:10Z

alphaswarm/services/cookiefun/cookiefun_client.py

+        return PagedAgentsResponse(
+            data=[AgentMetrics(**agent) for agent in data["data"]],
+            currentPage=data["currentPage"],
+            totalPages=data["totalPages"],
+            totalCount=data["totalCount"],
+        )


pydantic dataclasses should already handling this for you. This might work

Suggested change

return PagedAgentsResponse(

data=[AgentMetrics(**agent) for agent in data["data"]],

currentPage=data["currentPage"],

totalPages=data["totalPages"],

totalCount=data["totalCount"],

)

return PagedAgentsResponse(**data)

gkoch78 · 2025-01-31T19:09:06Z

tests/integration/conftest.py

+def cookiefun_client(default_config: Config) -> CookieFunClient:
+    """Create CookieFun client for testing"""
+    # Set test API key if not present
+    return CookieFunClient()


that would properly the API key to be set in CI

gkoch78

see inline comments

catch up with main

alphaswarm/services/cookiefun/cookiefun_client.py

aflament · 2025-01-31T19:51:16Z

alphaswarm/services/cookiefun/cookiefun_client.py

+    agent_name: str = Field(alias="agentName")
+    twitter_usernames: List[str] = Field(alias="twitterUsernames")
+    mindshare_delta_percent: float = Field(alias="mindshareDeltaPercent")
+    market_cap: float = Field(alias="marketCap")
+    market_cap_delta_percent: float = Field(alias="marketCapDeltaPercent")
+    price_delta_percent: float = Field(alias="priceDeltaPercent")
+    volume_24_hours: float = Field(alias="volume24Hours")
+    volume_24_hours_delta_percent: float = Field(alias="volume24HoursDeltaPercent")
+    holders_count: int = Field(alias="holdersCount")
+    holders_count_delta_percent: float = Field(alias="holdersCountDeltaPercent")
+    average_impressions_count: float = Field(alias="averageImpressionsCount")
+    average_impressions_count_delta_percent: float = Field(alias="averageImpressionsCountDeltaPercent")
+    average_engagements_count: float = Field(alias="averageEngagementsCount")
+    average_engagements_count_delta_percent: float = Field(alias="averageEngagementsCountDeltaPercent")
+    followers_count: int = Field(alias="followersCount")
+    smart_followers_count: int = Field(alias="smartFollowersCount")
+    top_tweets: List[Tweet] = Field(alias="topTweets")


can we group in a logical manner all those fields

something like
`
# Optional/fields with defaults after
# Basic agent info
agent_name: str = Field(alias="agentName")
twitter_usernames: List[str] = Field(alias="twitterUsernames")

# Market metrics market_cap: float = Field(alias="marketCap") market_cap_delta_percent: float = Field(alias="marketCapDeltaPercent") price_delta_percent: float = Field(alias="priceDeltaPercent") volume_24_hours: float = Field(alias="volume24Hours") volume_24_hours_delta_percent: float = Field(alias="volume24HoursDeltaPercent") # Mindshare metrics mindshare_delta_percent: float = Field(alias="mindshareDeltaPercent") # Holder metrics holders_count: int = Field(alias="holdersCount") holders_count_delta_percent: float = Field(alias="holdersCountDeltaPercent") # Engagement metrics average_impressions_count: float = Field(alias="averageImpressionsCount") average_impressions_count_delta_percent: float = Field(alias="averageImpressionsCountDeltaPercent") average_engagements_count: float = Field(alias="averageEngagementsCount") average_engagements_count_delta_percent: float = Field(alias="averageEngagementsCountDeltaPercent") # Social metrics followers_count: int = Field(alias="followersCount") smart_followers_count: int = Field(alias="smartFollowersCount") top_tweets: List[Tweet] = Field(alias="topTweets")

`

We could, but this is the order in api output schema. Wouldn't it be easier to reconcile dataclass with api for users if in same order?

alphaswarm/services/cookiefun/cookiefun_client.py

aflament · 2025-01-31T20:04:56Z

alphaswarm/services/cookiefun/cookiefun_client.py

+            logger.exception("Error fetching data from Cookie.fun")
+            raise
+
+    def _parse_agent_response(self, response_data: dict) -> AgentMetrics:


Suggested change

def _parse_agent_response(self, response_data: dict) -> AgentMetrics:

def _parse_agent_metrics_response(self, response_data: dict) -> AgentMetrics:

alphaswarm/services/cookiefun/cookiefun_client.py

aflament · 2025-01-31T20:07:39Z

alphaswarm/services/cookiefun/cookiefun_client.py

+            ApiException: If API request fails
+        """
+        if not 1 <= page_size <= 25:
+            raise ValueError("page_size must be between 1 and 25")


Suggested change

raise ValueError("page_size must be between 1 and 25")

raise ValueError(f"page_size must be between 1 and 25, got {page_size}")

alphaswarm/tools/cookie/cookie_metrics.py

catch up with main.

aflament

😔

catch up with main

* feat cookiefun api * feat cookiefun api Get Agents Paged functionality * refactor cookiefun api * add cookie tools to examples/terminal * add integration test for cookie fun api * feat cookie fun api by token address * add cookie fun metrics by token symbol * add conftest to cookie integration test * reformat code to fix linting * reformat imports using poetry * fix property names to be lower_snake_case * fix linting * fix linting * refactor method naming * refactor method naming * add CookieMetricsBySymbol as tool * add cookie_metrics to examples/terminal * fix linting * resolve merge conflict in examples/terminal * add default values to cookiefun_client * fix linting * fix linting * add default values to dataclasses * fix linting * Decouple trading strategy from core AlphaSwarmAgent. (#32) * momentum strategy analysis as agent specilization instead of tool (#27) * inital work on research agent * rename to momentum strategy agent; added price chg calculator tool * lint * Remove lab directory from git tracking * Implement telegram image sending * Refactor evm client (#28) * extract erc20contract * WIP signing transaction * lint * clean up * clean up * use EMVClient in Uniswap * clean up * clean up * clean up * clean up * clean up * clean up * clean up ExactInputSingleParams * Create test_v3_router_contract.py * more clean up * more clean up * more clean up * fix test * Install matplotlib * Progress * Fix output formatting (#35) * Introduce hints file * Tune instructions in SendTelegramNotificationTool * Use claude-3-5-sonnet-20241022 as a model * Rename token_set to token_name_to_address * Fix comments * Add manual test for swap (#40) * add a couple of integration test on swap to run manually * add ExecuteSwapTool in terminal * quick fix * Adding project README and basic examples (#26) * Update README * Introduce basic tutorial examples --------- Co-authored-by: Arnaud Flament <[email protected]> * Create image as part of forecasting tool * reorganize lab agent tools; cookie tool description update * fixed wrong substitution in system prompt * added trading to lab advisor agent * simplified momentum agent in lab * Feature uniswap more clean up (#42) * use TokenInfo to convert in uniswap v2 * clean up private key and wallet address in UniswapClientBase * clean up PoolContract * fix return type for ERC20Cotnract.get_balance * clean up * fix integration test * Fix Telegram RuntimeError('Event loop is closed') * simiplfy cookie page description * Fix Telegram RuntimeError('Event loop is closed') (#44) * Decouple Config from class constructors (#45) * Initial commit * Merge with main * Update * Code clean-up (#46) * Code clean-up * Update * Fix tests * fix tests * Feature more lint (#47) * lint * more lint * more lint * clean up * clean up * enable `warn_unused_ignores` (#48) * enable `warn_unused_ignores` * Update terminal.py * updated research system prompt * Feature introduce slippage class (#50) * Create Slippage class * Add Unit tests * Update * Feature improve CI (#49) * add test reports to CI * update test artifacts * Update python.yaml * Update python.yaml * lint * Update pyproject.toml * fix workflow * Feature fix decimal from float (#51) * Update sol.py * Update sol.py * Update sol.py * Feature cookie api (#22) * feat cookiefun api * feat cookiefun api Get Agents Paged functionality * refactor cookiefun api * add cookie tools to examples/terminal * add integration test for cookie fun api * feat cookie fun api by token address * add cookie fun metrics by token symbol * add conftest to cookie integration test * reformat code to fix linting * reformat imports using poetry * fix property names to be lower_snake_case * fix linting * fix linting * refactor method naming * refactor method naming * add CookieMetricsBySymbol as tool * add cookie_metrics to examples/terminal * fix linting * resolve merge conflict in examples/terminal * add default values to cookiefun_client * fix linting * fix linting * add default values to dataclasses * fix linting * add token details todefault * added cookie api key to github workflow * fix linting * fix linting * add typing * add cookiefun_client to conftest --------- Co-authored-by: AlexM369 <[email protected]> Co-authored-by: Ethan <[email protected]> * research agent system prompt tweaks * debugging * resolve merge errors --------- Co-authored-by: AlexM369 <[email protected]> Co-authored-by: Dmytro Nikolaiev <[email protected]> Co-authored-by: Guillaume Koch <[email protected]> Co-authored-by: david <[email protected]> Co-authored-by: Arnaud Flament <[email protected]> Co-authored-by: Arnaud Flament <[email protected]> Co-authored-by: AlexM369 <[email protected]>

AlexM369 added 10 commits January 31, 2025 11:51

feat cookiefun api

a46a48c

feat cookiefun api Get Agents Paged functionality

027cff6

refactor cookiefun api

26e3f54

add cookie tools to examples/terminal

54cfaeb

add integration test for cookie fun api

e4c04ee

feat cookie fun api by token address

660500e

add cookie fun metrics by token symbol

998d86e

add conftest to cookie integration test

c96563b

Merge branch 'main' into feature_cookie_api

f077248

catch up with main

reformat code to fix linting

a6d67a1

AlexM369 requested a review from gkoch78 January 31, 2025 18:58

reformat imports using poetry

656359b

gkoch78 reviewed Jan 31, 2025

View reviewed changes

gkoch78 requested changes Jan 31, 2025

View reviewed changes

AlexM369 added 4 commits January 31, 2025 14:39

fix property names to be lower_snake_case

a876d60

Merge branch 'main' into feature_cookie_api

54bfc77

catch up with main

fix linting

438695c

fix linting

ddec143

aflament requested changes Jan 31, 2025

View reviewed changes

AlexM369 added 8 commits January 31, 2025 15:40

refactor method naming

66dea4f

refactor method naming

c6fa098

add CookieMetricsBySymbol as tool

24eb48c

Merge branch 'main' into feature_cookie_api

6b30eaf

add cookie_metrics to examples/terminal

5f16725

fix linting

463cc72

Merge branch 'main' into feature_cookie_api

ace23b8

resolve merge conflict in examples/terminal

255faad

AlexM369 added 6 commits February 3, 2025 10:01

add default values to cookiefun_client

78a0cef

fix linting

6de4cb3

fix linting

e0c2557

add default values to dataclasses

70bff09

Merge branch 'main' into feature_cookie_api

8a80189

fix linting

7b5b243

AlexM369 requested review from aflament and gkoch78 February 3, 2025 15:21

AlexM369 and others added 4 commits February 6, 2025 12:34

add token details todefault

5753175

Merge branch 'main' of https://github.com/chain-ml/alphaswarm

f1ae8a3

catch up with main.

add hint to cookie page tool description

0b3dfb1

added cookie api key to github workflow

1d78060

aflament approved these changes Feb 6, 2025

View reviewed changes

gkoch78 approved these changes Feb 7, 2025

View reviewed changes

AlexM369 added 7 commits February 7, 2025 08:44

Merge branch 'main' of https://github.com/chain-ml/alphaswarm

dfa5718

catch up with main

Merge branch 'main' into feature_cookie_api

9250556

fix linting

7fa7f31

add cookie fun api key to git workflow

327932c

fix linting

581e804

add typing

558f2c1

add cookiefun_client to conftest

3264766

AlexM369 merged commit 0f40685 into main Feb 7, 2025
2 checks passed

AlexM369 deleted the feature_cookie_api branch February 7, 2025 14:02

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Feature cookie api #22

Feature cookie api #22

AlexM369 commented Jan 31, 2025

gkoch78 Jan 31, 2025

AlexM369 Feb 3, 2025

gkoch78 Jan 31, 2025

AlexM369 Feb 3, 2025

gkoch78 Jan 31, 2025

AlexM369 Feb 3, 2025

gkoch78 Jan 31, 2025

gkoch78 left a comment

aflament Jan 31, 2025

aflament Jan 31, 2025

AlexM369 Jan 31, 2025

aflament Jan 31, 2025

AlexM369 Jan 31, 2025

aflament Jan 31, 2025

AlexM369 Jan 31, 2025

aflament left a comment

	def _parse_agent_response(self, response_data: dict) -> AgentMetrics:
	def _parse_agent_metrics_response(self, response_data: dict) -> AgentMetrics:

	raise ValueError("page_size must be between 1 and 25")
	raise ValueError(f"page_size must be between 1 and 25, got {page_size}")

Feature cookie api #22

Feature cookie api #22

Conversation

AlexM369 commented Jan 31, 2025

Choose a reason for hiding this comment

Choose a reason for hiding this comment

Choose a reason for hiding this comment

Choose a reason for hiding this comment

Choose a reason for hiding this comment

Choose a reason for hiding this comment

Choose a reason for hiding this comment

gkoch78 left a comment

Choose a reason for hiding this comment

Choose a reason for hiding this comment

Choose a reason for hiding this comment

Choose a reason for hiding this comment

Choose a reason for hiding this comment

Choose a reason for hiding this comment

Choose a reason for hiding this comment

Choose a reason for hiding this comment

aflament left a comment

Choose a reason for hiding this comment