mirrors/puter
1
0
Fork 0
mirror of https://github.com/HeyPuter/puter synced 2024-11-14 22:06:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: add test_mode flag

Browse Source
This commit is contained in:
KernelDeimos 2024-08-03 20:56:21 -04:00 committed by Eric Dubé
parent 78fa77d920
commit 9a9bd5eaf0
3 changed files with 48 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
src/backend/src/modules/puterai/AWSTextractService.js
View File

@ -1,6 +1,7 @@
const { TextractClient, AnalyzeDocumentCommand, InvalidS3ObjectException } = require("@aws-sdk/client-textract");
const BaseService = require("../../services/BaseService");
const APIError = require("../../api/APIError");
class AWSTextractService extends BaseService {
_construct () {
@ -10,6 +11,23 @@ class AWSTextractService extends BaseService {
static IMPLEMENTS = {
['puter-ocr']: {
async recognize ({ source, test_mode }) {
if ( test_mode ) {
return {
blocks: [
{
type: 'text/textract:WORD',
confidence: 0.9999998807907104,
text: 'Hello',
},
{
type: 'text/puter:sample-output',
confidence: 1,
text: 'The test_mode flag is set to true. This is a sample output.',
},
]
};
}
const resp = await this.analyze_document(source);
// Simplify the response for common interface

27
src/backend/src/modules/puterai/OpenAICompletionService.js
View File

@ -20,7 +20,32 @@ class OpenAICompletionService extends BaseService {
static IMPLEMENTS = {
['puter-chat-completion']: {
async complete ({ messages, vision }) {
async complete ({ messages, test_mode }) {
if ( test_mode ) {
const { LoremIpsum } = require('lorem-ipsum');
const li = new LoremIpsum({
sentencesPerParagraph: {
max: 8,
min: 4
},
wordsPerSentence: {
max: 20,
min: 12
},
});
return {
"index": 0,
"message": {
"role": "assistant",
"content": li.generateParagraphs(
Math.floor(Math.random() * 3) + 1
),
},
"logprobs": null,
"finish_reason": "stop"
}
}
const model = 'gpt-4o';
return await this.complete(messages, {
model,

8
src/backend/src/modules/puterai/OpenAIImageGenerationService.js
View File

@ -19,10 +19,6 @@ class OpenAIImageGenerationService extends BaseService {
static IMPLEMENTS = {
['puter-image-generation']: {
async generate ({ prompt, test_mode }) {
const url = await this.generate(prompt, {
ratio: this.constructor.RATIO_SQUARE,
});
if ( test_mode ) {
return new TypedValue({
$: 'string:url:web',
@ -30,6 +26,10 @@ class OpenAIImageGenerationService extends BaseService {
}, 'https://puter-sample-data.puter.site/image_example.png');
}
const url = await this.generate(prompt, {
ratio: this.constructor.RATIO_SQUARE,
});
const image = new TypedValue({
$: 'string:url:web',
content_type: 'image'