just going to remove html2text since its not being used only

GooeyAI · Oct 12, 2023 · 73cf54f · 73cf54f
1 parent 0364246
commit 73cf54f
Show file tree

Hide file tree

Showing 4 changed files with 2 additions and 18 deletions.
diff --git a/daras_ai/text_format.py b/daras_ai/text_format.py
@@ -2,12 +2,11 @@
 
 import parse
 from glom import glom
-from html2text import html2text
 
 input_spec_parse_pattern = "{" * 5 + "}" * 5
 
 
-def daras_ai_format_str(format_str, variables, do_html2text=False):
+def daras_ai_format_str(format_str, variables):
     input_spec_results: list[parse.Result] = list(
         parse.findall(input_spec_parse_pattern, format_str)
     )
@@ -18,8 +17,6 @@ def daras_ai_format_str(format_str, variables, do_html2text=False):
             variable_value = ""
         else:
             variable_value = str(variable_value)
-        if do_html2text:
-            variable_value = html2text(variable_value)
         if isinstance(variable_value, str):
             variable_value = variable_value.strip()
         format_str = format_str.replace("{{" + spec + "}}", str(variable_value))

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -13,7 +13,6 @@ python-decouple = "^3.6"
 requests = "^2.28.1"
 glom = "^22.1.0"
 parse = "^1.19.0"
-html2text = "^2020.1.16"
 pandas = "^2.0.1"
 google-cloud-firestore = "^2.7.0"
 replicate = "^0.4.0"

diff --git a/recipes/LetterWriter.py b/recipes/LetterWriter.py
@@ -256,7 +256,6 @@ def run(self, state: dict) -> typing.Iterator[str | None]:
         input_prompt = daras_ai_format_str(
             format_str=request.input_prompt,
             variables=response_json,
-            do_html2text=request.strip_html_2_text,
         )
 
         state["generated_input_prompt"] = input_prompt