updates

utils/interp_space_utils.py

@@ -478,9 +478,13 @@ def identify_style_features(author_texts: list[str], author_names: list[str], ma
 
     {author_texts}
     """
+
+    print('==================>>>>>>>>>>')
+    print(prompt)
+    print('==================>>>>>>>>>>')
     def _make_call():
         response = client.chat.completions.create(
-            model="gpt-4o-mini",
+            model="gpt-4o",
             messages=[
                 {"role": "assistant", "content": "You are a forensic linguist who knows how to analyze linguistic and stylometric similarites between texts."},
                 {"role": "user", "content": prompt}

utils/llm_feat_utils.py

@@ -52,10 +52,12 @@ def generate_feature_spans(client, text: str, features: list[str]) -> str:
     Style Features:
     {features}
     """
+    print('==================>>>>>>>>>>')
+    print(prompt)
+    print('==================>>>>>>>>>>')
     response = client.chat.completions.create(
-        model="gpt-4o-mini",
-        messages=[{"role":"user","content":prompt}],
-        temperature=0.3,
+        model="gpt-4o",
+        messages=[{"role":"user","content":prompt}]
     )
     content = response.choices[0].message.content
     content = content.replace('```json', '').replace('```','')