Added option to provide RAG docs

Author: hrshdhgd

poetry.lock

@@ -13,7 +13,7 @@ python = ">=3.9,<4.0.0"
 oaklib = ">=0.5.0"
 PyGithub = "^2.3.0"
 setuptools = ">=70.1.1"
-llm-change-agent = {version = "^0.0.7", extras = ["llm"], optional = true}
+llm-change-agent = {version = "^0.0.8", extras = ["llm"], optional = true}
 
 
 [tool.poetry.group.dev.dependencies]

pyproject.toml

@@ -149,8 +149,12 @@ def main(verbose: int, quiet: bool):
 llm_model_option = click.option(
     "--model", type=click.Choice(ALL_AVAILABLE_MODELS), help="Model to use for generation."
 )
-
-
+rag_docs_option = click.option(
+    "--rag-docs",
+    multiple=True,
+    default=[],
+    help=f"Paths to the docs directories, URLs, or lowercased ontology names.",
+)
 @main.command()
 @repo_option
 @state_option
@@ -216,6 +220,7 @@ def get_labels(repo: str, token: str):
 @use_llm_option
 @llm_provider_option
 @llm_model_option
+@rag_docs_option
 def process_issue(
     input: str,
     repo: str,
@@ -230,6 +235,7 @@ def process_issue(
     use_llm: bool = False,
     provider: str = None,
     model: str = None,
+    rag_docs: tuple[str] = None,
 ):
     """Run processes based on issue label.
 
@@ -281,6 +287,7 @@ def process_issue(
                     ctx.params["prompt"] = issue[BODY]
                     ctx.params["provider"] = provider
                     ctx.params["model"] = model
+                    ctx.params["docs"] = rag_docs
                     response = extract_commands(execute.invoke(ctx))
                     KGCL_COMMANDS = [
                         command.replace('"', "'") for command in ast.literal_eval(response)