ros/claude-task-master
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge remote-tracking branch 'origin/next' into telemetry

Browse Source
This commit is contained in:
Ralph Khreish
2025-05-16 18:16:58 +02:00
parent da636f6681 da317f2607
commit bc19bc7927
40 changed files with 387 additions and 229 deletions
Download Patch File Download Diff File Expand all files Collapse all files

60
src/ai-providers/google.js
View File

@@ -12,6 +12,16 @@ import { log } from '../../scripts/modules/utils.js'; // Import logging utility
const DEFAULT_MODEL = 'gemini-2.5-pro-exp-03-25'; // Or a suitable default
const DEFAULT_TEMPERATURE = 0.2; // Or a suitable default
function getClient(apiKey, baseUrl) {
if (!apiKey) {
throw new Error('Google API key is required.');
}
return createGoogleGenerativeAI({
apiKey: apiKey,
...(baseUrl && { baseURL: baseUrl })
});
}
/**
* Generates text using a Google AI model.
*
@@ -29,7 +39,8 @@ async function generateGoogleText({
modelId = DEFAULT_MODEL,
temperature = DEFAULT_TEMPERATURE,
messages,
maxTokens // Note: Vercel SDK might handle this differently, needs verification
maxTokens,
baseUrl
}) {
if (!apiKey) {
throw new Error('Google API key is required.');
@@ -37,18 +48,13 @@ async function generateGoogleText({
log('info', `Generating text with Google model: ${modelId}`);
try {
// const google = new GoogleGenerativeAI({ apiKey }); // Incorrect instantiation
const googleProvider = createGoogleGenerativeAI({ apiKey }); // Correct instantiation
// const model = google.getGenerativeModel({ model: modelId }); // Incorrect model retrieval
const model = googleProvider(modelId); // Correct model retrieval
// Construct payload suitable for Vercel SDK's generateText
// Note: The exact structure might depend on how messages are passed
const googleProvider = getClient(apiKey, baseUrl);
const model = googleProvider(modelId);
const result = await generateText({
model, // Pass the model instance
messages, // Pass the messages array directly
model,
messages,
temperature,
maxOutputTokens: maxTokens // Map to correct Vercel SDK param if available
maxOutputTokens: maxTokens
});
// Assuming result structure provides text directly or within a property
@@ -66,7 +72,7 @@ async function generateGoogleText({
'error',
`Error generating text with Google (${modelId}): ${error.message}`
);
throw error; // Re-throw for unified service handler
throw error;
}
}
@@ -87,7 +93,8 @@ async function streamGoogleText({
modelId = DEFAULT_MODEL,
temperature = DEFAULT_TEMPERATURE,
messages,
maxTokens
maxTokens,
baseUrl
}) {
if (!apiKey) {
throw new Error('Google API key is required.');
@@ -95,19 +102,15 @@ async function streamGoogleText({
log('info', `Streaming text with Google model: ${modelId}`);
try {
// const google = new GoogleGenerativeAI({ apiKey }); // Incorrect instantiation
const googleProvider = createGoogleGenerativeAI({ apiKey }); // Correct instantiation
// const model = google.getGenerativeModel({ model: modelId }); // Incorrect model retrieval
const model = googleProvider(modelId); // Correct model retrieval
const googleProvider = getClient(apiKey, baseUrl);
const model = googleProvider(modelId);
const stream = await streamText({
model, // Pass the model instance
model,
messages,
temperature,
maxOutputTokens: maxTokens
});
return stream; // Return the stream directly
return stream;
} catch (error) {
log(
'error',
@@ -138,7 +141,8 @@ async function generateGoogleObject({
messages,
schema,
objectName, // Note: Vercel SDK might use this differently or not at all
maxTokens
maxTokens,
baseUrl
}) {
if (!apiKey) {
throw new Error('Google API key is required.');
@@ -146,20 +150,14 @@ async function generateGoogleObject({
log('info', `Generating object with Google model: ${modelId}`);
try {
// const google = new GoogleGenerativeAI({ apiKey }); // Incorrect instantiation
const googleProvider = createGoogleGenerativeAI({ apiKey }); // Correct instantiation
// const model = google.getGenerativeModel({ model: modelId }); // Incorrect model retrieval
const model = googleProvider(modelId); // Correct model retrieval
const googleProvider = getClient(apiKey, baseUrl);
const model = googleProvider(modelId);
const result = await generateObject({
model, // Pass the model instance
model,
schema,
messages,
temperature,
maxOutputTokens: maxTokens
// Note: 'objectName' or 'mode' might not be directly applicable here
// depending on how `@ai-sdk/google` handles `generateObject`.
// Check SDK docs if specific tool calling/JSON mode needs explicit setup.
});
// return object; // Return the parsed object