make ai sdk native #698
make ai sdk native #698
Conversation
🦋 Changeset detectedLatest commit: 1848be5 The changes in this PR will be included in the next version bump. This PR includes changesets to release 1 package
Not sure what this means? Click here to learn what changesets are. Click here if you're a maintainer who wants to add another changeset to this PR |
|
I started looking into this yesterday; it seems like if we're going with aisdk we should probably change our internal inference logic to adapt to |
* avoid using model-provider map * prettier * remove duplicate dependency
![greptile-apps[bot]](https://avatars.githubusercontent.com/in/867647?s=60&v=4){kind=small}
There was a problem hiding this comment.
PR Summary
This PR integrates the AI SDK as a first-class LLM client in Stagehand, moving from an external implementation to a native one with enhanced logging and caching capabilities.
- Added
lib/llm/aisdk.tsimplementingAISdkClientwith support for multiple AI providers (OpenAI, Google, Anthropic, Groq, Cerebras) - Modified model naming convention to use format
aisdk/provider/model(e.g.aisdk/openai/gpt-4o) for simplified integration - Moved AI SDK dependencies from devDependencies to optionalDependencies in
package.json, allowing selective provider installation - Added support for provider-specific API keys in environment variables through
lib/index.ts - Enhanced
types/model.tsto support flexible model strings while maintaining type safety
10 file(s) reviewed, 8 comment(s)
Edit PR Review Bot Settings | Greptile
| @@ -34,14 +34,15 @@ export const AvailableModelSchema = z.enum([ | |||
| "gemini-2.5-pro-preview-03-25", | |||
| ]); | |||
|
|
|||
| export type AvailableModel = z.infer<typeof AvailableModelSchema>; | |||
| export type AvailableModel = z.infer<typeof AvailableModelSchema> | string; | |||
There was a problem hiding this comment.
style: Adding string to AvailableModel type weakens type safety. Consider using a branded type or maintaining an explicit list of supported models to prevent runtime errors from invalid model names.
why
what changed
test plan