fill.ts 398 B

1234567891011121314151617181920
  1. import ollama from 'ollama'
  2. const prefix = `def remove_non_ascii(s: str) -> str:
  3. """
  4. `
  5. const suffix = `
  6. return result
  7. `
  8. const response = await ollama.generate({
  9. model: 'codellama:7b-code',
  10. prompt: `<PRE> ${prefix} <SUF>${suffix} <MID>`,
  11. options: {
  12. num_predict: 128,
  13. temperature: 0,
  14. top_p: 0.9,
  15. presence_penalty: 0,
  16. stop: ['<EOT>'],
  17. },
  18. })
  19. console.log(response.response)