add

agentgraph/testing/knowledge_graph_tester.py

@@ -49,9 +49,9 @@ if LANGFUSE_SECRET_KEY:
 if LANGFUSE_PUBLIC_KEY and LANGFUSE_SECRET_KEY:
     os.environ["OTEL_EXPORTER_OTLP_ENDPOINT"] = f"{LANGFUSE_HOST}/api/public/otel"
     os.environ["OTEL_EXPORTER_OTLP_HEADERS"] = f"Authorization=Basic {LANGFUSE_AUTH}"
-import openlit
-
-openlit.init()
+# Commented out to reduce verbose telemetry output
+# import openlit
+# openlit.init()
 
 # Only set environment variable if OPENAI_API_KEY is not None
 if OPENAI_API_KEY:

agentgraph/testing/perturbation_types/jailbreak.py

@@ -43,9 +43,9 @@ def load_jailbreak_techniques() -> List[Dict[str, Any]]:
         techniques = []
         for _, row in df.iterrows():
             techniques.append({
-                'name': row.get('topic', 'Unknown'),
-                'prompt': row.get('prompt', ''),
-                'description': row.get('source', '')
+                'name': row['topic'] if 'topic' in row else 'Unknown',
+                'prompt': row['prompt'] if 'prompt' in row else '',
+                'description': row['source'] if 'source' in row else ''
             })
         
         logger.info(f"Successfully loaded {len(techniques)} jailbreak techniques from CSV file")