File size: 756 Bytes
2f7798c
c32ec0d
2f7798c
c32ec0d
2f7798c
c32ec0d
 
 
 
 
2f7798c
 
 
c32ec0d
 
 
 
 
 
 
 
 
 
 
 
2f7798c
c32ec0d
2f7798c
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
import { LLMResponse } from "@/types"
import { cleanJson } from "./cleanJson"

export function dirtyLLMJsonParser(input: string): LLMResponse {

  if (input.includes("```")) {
    input = input.split("```")[0]
  }
  // we only keep what's after the first [
  let jsonOrNot = cleanJson(input)

  const jsonData = JSON.parse(jsonOrNot) as LLMResponse

  const results = jsonData.map((item, i) => {
    let panel = i
    let caption = item.caption ? item.caption.trim() : ''
    let instructions = item.instructions ? item.instructions.trim() : ''
    if (!instructions && caption) {
      instructions = caption
    } 
    if (!caption && instructions) {
      caption = instructions
    }
    return { panel, caption, instructions }
  })

  return results
}