Spaces:
Running
Running
File size: 3,298 Bytes
c232e44 0d6f04b f665131 0d6f04b 63d0776 c232e44 a1c5622 eae2367 a1c5622 c232e44 4af6326 c232e44 4af6326 c232e44 11e3c5e c232e44 11e3c5e c232e44 11e3c5e c232e44 ae074fc c232e44 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 |
import toast from 'react-hot-toast';
const ANSWERS_PREFIX = 'answers';
export const generateAnswersImageMarkdown = (index: number, url: string) => {
return ``;
};
export const cleanInputMessage = (content: string) => {
return content
.replace(/!\[input-.*?\)/g, '')
.replace(/<video[^>]*>.*?<\/video>/g, '');
};
export const cleanAnswerMessage = (content: string) => {
return content.replace(/!\[answers.*?\.png\)/g, '');
};
type PlansBody =
| {
type: 'plans';
status: 'started';
}
| {
type: 'plans';
status: 'completed';
payload: Array<Record<string, string>>;
};
type ToolsBody =
| {
type: 'tools';
status: 'started';
}
| {
type: 'tools';
status: 'completed';
payload: Array<Record<string, string>>;
};
type CodeBody =
| {
type: 'code';
status: 'started';
}
| {
type: 'code';
status: 'running';
payload: {
code: string;
test: string;
};
}
| {
type: 'code';
status: 'completed' | 'failed';
payload: {
code: string;
test: string;
result: string;
};
};
// this will return if self_reflection flag is true
type ReflectionBody =
| {
type: 'self_reflection';
status: 'started';
}
| {
type: 'self_reflection';
status: 'completed' | 'failed';
payload: { feedback: string; success: boolean };
};
export type CodeResult = {
code: string;
test: string;
result: string;
};
export type ChunkBody =
| {
type: 'plans' | 'tools' | 'code' | 'final_code' | 'final_error';
status: 'started' | 'completed' | 'failed' | 'running';
payload:
| Array<Record<string, string>> // PlansBody | ToolsBody
| CodeResult; // CodeBody
}
| PrismaJson.FinalChatResult;
/**
* Formats the stream logs and returns an array of grouped sections.
*
* @param content - The content of the stream logs.
* @returns An array of grouped sections and an optional final code result.
*/
export const formatStreamLogs = (
content: string | null | undefined,
): [ChunkBody[], CodeResult?] => {
if (!content) return [[], undefined];
const streamLogs = content.split('\n').filter(log => !!log);
const buffer = streamLogs.pop();
const parsedStreamLogs: ChunkBody[] = [];
try {
streamLogs.forEach(streamLog =>
parsedStreamLogs.push(JSON.parse(streamLog)),
);
} catch {
toast.error('Error parsing stream logs');
return [[], undefined];
}
if (buffer) {
try {
const lastLog = JSON.parse(buffer);
parsedStreamLogs.push(lastLog);
} catch {
console.log(buffer);
}
}
// Merge consecutive logs of the same type to the latest status
const groupedSections = parsedStreamLogs.reduce((acc, curr) => {
if (
acc.length > 0 &&
acc[acc.length - 1].type === curr.type &&
curr.status !== 'started'
) {
acc[acc.length - 1] = curr;
} else {
acc.push(curr);
}
return acc;
}, [] as ChunkBody[]);
return [
groupedSections.filter(section => section.type !== 'final_code'),
groupedSections.find(section => section.type === 'final_code')
?.payload as CodeResult,
];
};
|