fix init

hegelai · Mar 9, 2024 · 412e36b · 412e36b
1 parent 8276bca
commit 412e36b
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/prompttools/utils/__init__.py b/prompttools/utils/__init__.py
@@ -12,8 +12,9 @@
 from .autoeval_with_docs import autoeval_with_documents
 from .chunk_text import chunk_text
 from .expected import compute_similarity_against_model
+from .moderation import apply_moderation
 from .ranking_correlation import ranking_correlation
-from .similarity import cos_similarity, semantic_similarity, structural_similarity
+from .similarity import semantic_similarity, cos_similarity
 from .validate_json import validate_json_response
 from .validate_python import validate_python_response
 
@@ -26,13 +27,13 @@
  "chunk_text",
  "compute_similarity_against_model",
  "expected",
+ "apply_moderation",
+ "ranking_correlation",
+ "semantic_similarity",
+ "cos_similarity",
+ "similarity",
  "validate_json",
  "validate_json_response",
  "validate_python",
  "validate_python_response",
- "ranking_correlation",
- "cos_similarity",
- "semantic_similarity",
- "structural_similarity",
- "similarity",
-]
+]