predefinedschema.py 807 B

12345678910111213141516171819202122232425262728293031
  1. import requests
  2. import json
  3. import random
  4. model = "llama3.2"
  5. template = {
  6. "firstName": "",
  7. "lastName": "",
  8. "address": {
  9. "street": "",
  10. "city": "",
  11. "state": "",
  12. "zipCode": ""
  13. },
  14. "phoneNumber": ""
  15. }
  16. prompt = f"generate one realistically believable sample data set of a persons first name, last name, address in the US, and phone number. \nUse the following template: {json.dumps(template)}."
  17. data = {
  18. "prompt": prompt,
  19. "model": model,
  20. "format": "json",
  21. "stream": False,
  22. "options": {"temperature": 2.5, "top_p": 0.99, "top_k": 100},
  23. }
  24. print(f"Generating a sample user")
  25. response = requests.post("http://localhost:11434/api/generate", json=data, stream=False)
  26. json_data = json.loads(response.text)
  27. print(json.dumps(json.loads(json_data["response"]), indent=2))