Skip to content

Commit b96b96a

Browse files
authored
Merge pull request #1 from tomguluson92/tomguluson92-patch-1
Update generation.ts
2 parents a9d8417 + f5c0bea commit b96b96a

File tree

1 file changed

+7
-5
lines changed

1 file changed

+7
-5
lines changed

packages/core/src/generation.ts

+7-5
Original file line numberDiff line numberDiff line change
@@ -113,6 +113,7 @@ export async function generateText({
113113
);
114114

115115
switch (provider) {
116+
// OPENAI & LLAMACLOUD shared same structure.
116117
case ModelProviderName.OPENAI:
117118
case ModelProviderName.LLAMACLOUD: {
118119
elizaLogger.debug("Initializing OpenAI model.");
@@ -139,7 +140,7 @@ export async function generateText({
139140
case ModelProviderName.GOOGLE: {
140141
const google = createGoogleGenerativeAI();
141142

142-
const { text: anthropicResponse } = await aiGenerateText({
143+
const { text: googleResponse } = await aiGenerateText({
143144
model: google(model),
144145
prompt: context,
145146
system:
@@ -152,7 +153,8 @@ export async function generateText({
152153
presencePenalty: presence_penalty,
153154
});
154155

155-
response = anthropicResponse;
156+
response = googleResponse;
157+
elizaLogger.debug("Received response from Google model.");
156158
break;
157159
}
158160

@@ -280,7 +282,7 @@ export async function generateText({
280282
const serverUrl = models[provider].endpoint;
281283
const openai = createOpenAI({ apiKey, baseURL: serverUrl });
282284

283-
const { text: openaiResponse } = await aiGenerateText({
285+
const { text: redpillResponse } = await aiGenerateText({
284286
model: openai.languageModel(model),
285287
prompt: context,
286288
temperature: temperature,
@@ -293,8 +295,8 @@ export async function generateText({
293295
presencePenalty: presence_penalty,
294296
});
295297

296-
response = openaiResponse;
297-
elizaLogger.debug("Received response from OpenAI model.");
298+
response = redpillResponse;
299+
elizaLogger.debug("Received response from redpill model.");
298300
break;
299301
}
300302

0 commit comments

Comments
 (0)