Merge pull request #223 from mw2000/mw2000/twitter-toolkit

feat: Twitter toolkit implementation

Author: jasonhedman

next.config.js

@@ -22,6 +22,7 @@ const config = {
   experimental: {
     authInterrupts: true,
   },
+  serverExternalPackages: ["twitter-api-v2"],
 };
 
 export default config;

package.json

@@ -39,7 +39,7 @@
     "@discordjs/rest": "^2.5.1",
     "@e2b/code-interpreter": "^1.5.1",
     "@hookform/resolvers": "^5.2.0",
-    "@icons-pack/react-simple-icons": "^13.1.0",
+    "@icons-pack/react-simple-icons": "^13.7.0",
     "@llm-ui/code": "^0.13.3",
     "@llm-ui/markdown": "^0.13.3",
     "@llm-ui/react": "^0.13.3",
@@ -113,6 +113,7 @@
     "strava": "^2.3.0",
     "superjson": "^2.2.1",
     "tailwind-merge": "^3.3.0",
+    "twitter-api-v2": "^1.24.0",
     "usehooks-ts": "^3.1.1",
     "uuid": "^11.1.0",
     "vaul": "^1.1.2",

pnpm-lock.yaml

@@ -17,6 +17,7 @@ import { discordClientToolkit } from "./discord/client";
 import { stravaClientToolkit } from "./strava/client";
 import { spotifyClientToolkit } from "./spotify/client";
 import { videoClientToolkit } from "./video/client";
+import { twitterClientToolkit } from "./twitter/client";
 
 export type ClientToolkits = {
   [K in Toolkits]: ClientToolkit<
@@ -38,6 +39,7 @@ export const clientToolkits: ClientToolkits = {
   [Toolkits.Strava]: stravaClientToolkit,
   [Toolkits.Spotify]: spotifyClientToolkit,
   [Toolkits.Video]: videoClientToolkit,
+  [Toolkits.Twitter]: twitterClientToolkit,
 };
 
 export function getClientToolkit<T extends Toolkits>(

src/toolkits/toolkits/client.ts

@@ -11,6 +11,7 @@ import { discordToolkitServer } from "./discord/server";
 import { stravaToolkitServer } from "./strava/server";
 import { spotifyToolkitServer } from "./spotify/server";
 import { videoToolkitServer } from "./video/server";
+import { twitterToolkitServer } from "./twitter/server";
 import {
   Toolkits,
   type ServerToolkitNames,
@@ -37,6 +38,7 @@ export const serverToolkits: ServerToolkits = {
   [Toolkits.Strava]: stravaToolkitServer,
   [Toolkits.Spotify]: spotifyToolkitServer,
   [Toolkits.Video]: videoToolkitServer,
+  [Toolkits.Twitter]: twitterToolkitServer,
 };
 
 export function getServerToolkit<T extends Toolkits>(

src/toolkits/toolkits/server.ts

@@ -22,6 +22,8 @@ import type { spotifyParameters } from "./spotify/base";
 import type { SpotifyTools } from "./spotify/tools";
 import type { VideoTools } from "./video/tools";
 import type { videoParameters } from "./video/base";
+import type { TwitterTools } from "./twitter/tools";
+import type { twitterParameters } from "./twitter/base";
 
 export enum Toolkits {
   Exa = "exa",
@@ -36,6 +38,7 @@ export enum Toolkits {
   Strava = "strava",
   Spotify = "spotify",
   Video = "video",
+  Twitter = "twitter",
 }
 
 export type ServerToolkitNames = {
@@ -51,6 +54,7 @@ export type ServerToolkitNames = {
   [Toolkits.Strava]: StravaTools;
   [Toolkits.Spotify]: SpotifyTools;
   [Toolkits.Video]: VideoTools;
+  [Toolkits.Twitter]: TwitterTools;
 };
 
 export type ServerToolkitParameters = {
@@ -66,4 +70,5 @@ export type ServerToolkitParameters = {
   [Toolkits.Strava]: typeof stravaParameters.shape;
   [Toolkits.Spotify]: typeof spotifyParameters.shape;
   [Toolkits.Video]: typeof videoParameters.shape;
+  [Toolkits.Twitter]: typeof twitterParameters.shape;
 };

src/toolkits/toolkits/shared.ts

@@ -0,0 +1,17 @@
+import type { ToolkitConfig } from "@/toolkits/types";
+import { z } from "zod";
+import { TwitterTools } from "./tools";
+import { getUserProfileTool, getLatestTweetsTool } from "./tools";
+
+export const twitterParameters = z.object({});
+
+export const baseTwitterToolkitConfig: ToolkitConfig<
+  TwitterTools,
+  typeof twitterParameters.shape
+> = {
+  tools: {
+    [TwitterTools.GetUserProfile]: getUserProfileTool,
+    [TwitterTools.GetLatestTweets]: getLatestTweetsTool,
+  },
+  parameters: twitterParameters,
+};

src/toolkits/toolkits/twitter/base.ts

@@ -0,0 +1,51 @@
+import { SiX } from "@icons-pack/react-simple-icons";
+
+import { createClientToolkit } from "@/toolkits/create-toolkit";
+
+import { baseTwitterToolkitConfig } from "./base";
+import { TwitterTools } from "./tools";
+import {
+  getUserProfileToolConfigClient,
+  getLatestTweetsToolConfigClient,
+} from "./tools/client";
+
+import { ToolkitGroups } from "@/toolkits/types";
+
+import { TwitterWrapper } from "./wrapper";
+
+export const twitterClientToolkit = createClientToolkit(
+  baseTwitterToolkitConfig,
+  {
+    name: "Twitter",
+    description: "Get Twitter user profiles and latest tweets",
+    icon: SiX,
+    form: null,
+    Wrapper: TwitterWrapper,
+    type: ToolkitGroups.DataSource,
+    envVars: [
+      {
+        type: "all",
+        keys: ["AUTH_TWITTER_ID", "AUTH_TWITTER_SECRET"],
+        description: (
+          <span>
+            Get OAuth credentials from{" "}
+            <a
+              href="https://developer.twitter.com/en/portal/dashboard"
+              target="_blank"
+              rel="noopener noreferrer"
+              className="text-blue-500 underline hover:text-blue-700"
+            >
+              Twitter Developer Portal
+            </a>
+            . You&apos;ll need to create an app and generate OAuth 1.0a
+            credentials.
+          </span>
+        ),
+      },
+    ],
+  },
+  {
+    [TwitterTools.GetUserProfile]: getUserProfileToolConfigClient,
+    [TwitterTools.GetLatestTweets]: getLatestTweetsToolConfigClient,
+  },
+);

src/toolkits/toolkits/twitter/client.tsx

@@ -0,0 +1,51 @@
+import { createServerToolkit } from "@/toolkits/create-toolkit";
+import { baseTwitterToolkitConfig } from "./base";
+import {
+  getUserProfileToolConfigServer,
+  getLatestTweetsToolConfigServer,
+} from "./tools/server";
+import { TwitterTools } from "./tools";
+import { api } from "@/trpc/server";
+import { TwitterApi } from "twitter-api-v2";
+
+export const twitterToolkitServer = createServerToolkit(
+  baseTwitterToolkitConfig,
+  `You have access to the Twitter toolkit for fetching user profiles and tweets. This toolkit provides:
+
+- **Get User Profile**: Retrieve detailed information about a Twitter user by their username
+- **Get Latest Tweets**: Fetch the most recent tweets from a user (up to 100 tweets)
+
+**Tool Sequencing Strategies:**
+1. **User Research**: Start with Get User Profile to understand a user's background, then use Get Latest Tweets to see their recent activity
+2. **Content Analysis**: Use Get Latest Tweets to analyze a user's recent posts and engagement patterns
+3. **Profile Verification**: Combine Get User Profile with Get Latest Tweets to verify user authenticity and activity
+
+**Best Practices:**
+- Use usernames without the @ symbol (e.g., 'elonmusk' not '@elonmusk')
+- For tweet analysis, start with fewer results (10-20) to get a quick overview
+- Use the exclude_retweets and exclude_replies options to focus on original content
+- Consider the user's follower count and verification status when analyzing their influence`,
+  async () => {
+    const account = await api.accounts.getAccountByProvider("twitter");
+
+    if (!account) {
+      throw new Error(
+        "No Twitter account found. Please connect your Twitter account first.",
+      );
+    }
+
+    if (!account.access_token) {
+      throw new Error(
+        "Twitter access token not found. Please reconnect your Twitter account.",
+      );
+    }
+
+    // Create Twitter API client with user's access token
+    const client = new TwitterApi(account.access_token);
+
+    return {
+      [TwitterTools.GetUserProfile]: getUserProfileToolConfigServer(client),
+      [TwitterTools.GetLatestTweets]: getLatestTweetsToolConfigServer(client),
+    };
+  },
+);

src/toolkits/toolkits/twitter/server.ts

@@ -0,0 +1,2 @@
+export * from "./profile/client";
+export * from "./tweets/client";