retriever

Attributes

log

Classes

JanusRetriever
ActiveUsingsRetriever
TextSearchRetriever
LanguageDocsRetriever

Module Contents

retriever.log

class retriever.JanusRetriever

Bases: langchain_core.runnables.Runnable

invoke(input, config=None, **kwargs)

Parameters:

• input (janus.language.block.CodeBlock) –

• config (langchain_core.runnables.RunnableConfig | None) –

Return type:

dict

get_context(code_block)

Parameters:

code_block (janus.language.block.CodeBlock) –

Return type:

str

class retriever.ActiveUsingsRetriever

Bases: JanusRetriever

get_context(code_block)

Parameters:

code_block (janus.language.block.CodeBlock) –

Return type:

str

class retriever.TextSearchRetriever(retriever)

Bases: JanusRetriever

Parameters:

retriever (langchain_core.retrievers.BaseRetriever) –

retriever: langchain_core.retrievers.BaseRetriever

get_context(code_block)

Parameters:

code_block (janus.language.block.CodeBlock) –

Return type:

str

class retriever.LanguageDocsRetriever(llm, language_name, prompt_template_name='retrieval/language_docs')

Bases: JanusRetriever

Parameters:

• llm (janus.llm.models_info.JanusModel) –

• language_name (str) –

• prompt_template_name (str) –

get_context(code_block)

Parameters:

code_block (janus.language.block.CodeBlock) –

Return type:

str