chat comple

metadata_transformer.py

@@ -15,13 +15,13 @@ You are a metadata schema translator. You translate metadata from one schema to
 
 def translate(schema_input, schema_target):
     prompt = '"""{} \n Translate the schema metadata file above to the schema: {}"""'.format(schema_input, schema_target)    
-    response = huggingface.Completion.create(
+    response = huggingface.ChatCompletion.create(
         model="meta-llama/Llama-2-7b-hf",
-        # messages=[
-        #     {"role": "system", "content": system_message},
-        #     {"role": "user", "content": prompt},
-        # ],
-        prompt=prompt,
+        messages=[
+            {"role": "system", "content": system_message},
+            {"role": "user", "content": prompt},
+        ],
+        # prompt=prompt,
         temperature=0.9,
         top_p=0.6,
         max_tokens=256,