From 6dbef138376b76a73ecee5abdfb1f921f1a60101 Mon Sep 17 00:00:00 2001
From: you-n-g <you-n-g@users.noreply.github.com>
Date: Fri, 4 Jul 2025 20:31:06 +0800
Subject: [PATCH] chore: document APIBackend parameters and remove sparse dep
 (#1019)

---
 rdagent/oai/backend/base.py | 12 ++++++++++++
 requirements.txt            |  3 ---
 2 files changed, 12 insertions(+), 3 deletions(-)

diff --git a/rdagent/oai/backend/base.py b/rdagent/oai/backend/base.py
index e9ef0880..049a34e3 100644
--- a/rdagent/oai/backend/base.py
+++ b/rdagent/oai/backend/base.py
@@ -279,6 +279,18 @@ class APIBackend(ABC):
         Responseible for building messages and logging messages
 
         TODO: What is weird is that the function is called before we seperate embeddings and chat completion.
+
+        Parameters
+        ----------
+        user_prompt : str
+        system_prompt : str | None
+        former_messages : list | None
+        response_format : BaseModel | dict
+            A BaseModel based on pydantic or a dict
+        **kwargs
+        Returns
+        -------
+        str
         """
         if former_messages is None:
             former_messages = []
diff --git a/requirements.txt b/requirements.txt
index a4cdec44..5a8d1e9c 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -66,6 +66,3 @@ genson
 mlflow
 azureml-mlflow
 types-pytz
-
-# data science scenario for custom dataset
-sparse
\ No newline at end of file