janus.converter.diagram#

Attributes#

log

Classes#

DiagramGenerator

DiagramGenerator

Module Contents#

janus.converter.diagram.log#
class janus.converter.diagram.DiagramGenerator(diagram_type='Activity', add_documentation=False, **kwargs)#

Bases: janus.converter.document.Documenter

DiagramGenerator

A class that translates code from one programming language to a set of diagrams.

Initialize the DiagramGenerator class

Parameters:

  • model – The LLM to use for translation. If an OpenAI model, the OPENAI_API_KEY environment variable must be set and the OPENAI_ORG_ID environment variable should be set if needed.

  • model_arguments – Additional arguments to pass to the LLM constructor.

  • source_language – The source programming language.

  • max_prompts – The maximum number of prompts to try before giving up.

  • db_path – path to chroma database

  • db_config – database configuraiton

  • diagram_type – type of PLANTUML diagram to generate