janus-llm 3.5.0__tar.gz → 3.5.1__tar.gz

{janus_llm-3.5.0 → janus_llm-3.5.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: janus-llm
-Version: 3.5.0
+Version: 3.5.1
 Summary: A transcoding library using LLMs.
 Home-page: https://github.com/janus-llm/janus-llm
 License: Apache 2.0

{janus_llm-3.5.0 → janus_llm-3.5.1}/janus/__init__.py

@@ -5,7 +5,7 @@ from langchain_core._api.deprecation import LangChainDeprecationWarning
 from janus.converter.translate import Translator
 from janus.metrics import *  # noqa: F403
 
-__version__ = "3.5.0"
+__version__ = "3.5.1"
 
 # Ignoring a deprecation warning from langchain_core that I can't seem to hunt down
 warnings.filterwarnings("ignore", category=LangChainDeprecationWarning)

{janus_llm-3.5.0 → janus_llm-3.5.1}/janus/converter/diagram.py

@@ -130,7 +130,7 @@ class DiagramGenerator(Documenter):
         If the relevant fields have not been changed since the last time this method was
         called, nothing happens.
         """
-        self._diagram_prompt_engine = MODEL_PROMPT_ENGINES[self._model_name](
+        self._diagram_prompt_engine = MODEL_PROMPT_ENGINES[self._model_id](
             source_language=self._source_language,
             target_language="text",
             target_version=None,

{janus_llm-3.5.0 → janus_llm-3.5.1}/janus/converter/document.py

@@ -125,7 +125,7 @@ class MadLibsDocumenter(Documenter):
             out_text = self._parser.parse(working_block.text)
             obj.update(json.loads(out_text))
 
-        self._parser.set_reference(block.original)
+        self._parser.parse_input(block.original)
         block.text = self._parser.parse(json.dumps(obj))
         block.tokens = self._llm.get_num_tokens(block.text)
         block.translated = True

{janus_llm-3.5.0 → janus_llm-3.5.1}/janus/parsers/doc_parser.py

@@ -92,11 +92,12 @@ class MadlibsDocumentationParser(JanusParser):
     def __init__(self):
         super().__init__(expected_keys=[])
 
-    def parse_input(self, block: CodeBlock):
+    def parse_input(self, block: CodeBlock) -> str:
         # TODO: Perform comment stripping/placeholding here rather than in script
         text = super().parse_input(block)
         comment_ids = re.findall(r"<(?:BLOCK|INLINE)_COMMENT (\w{8})>", text)
         self.expected_keys = set(comment_ids)
+        return text
 
     def parse(self, text: str | BaseMessage) -> str:
         if isinstance(text, BaseMessage):

{janus_llm-3.5.0 → janus_llm-3.5.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "janus-llm"
-version = "3.5.0"
+version = "3.5.1"
 description = "A transcoding library using LLMs."
 authors = ["Michael Doyle <mdoyle@mitre.org>", "Chris Glasz <cglasz@mitre.org>",
            "Chris Tohline <ctohline@mitre.org>", "William Macke <wmacke@mitre.org>",