diff --git a/pr_agent/tools/pr_similar_issue.py b/pr_agent/tools/pr_similar_issue.py index 7a97d85b..5b276759 100644 --- a/pr_agent/tools/pr_similar_issue.py +++ b/pr_agent/tools/pr_similar_issue.py @@ -438,7 +438,7 @@ class PRSimilarIssue: level=IssueLevel.COMMENT) ) corpus.append(comment_record) - df = pd.DataFrame(corpus.dict()["documents"]) + df = pd.DataFrame(corpus.model_dump()["documents"]) get_logger().info('Done') get_logger().info('Embedding...') @@ -534,7 +534,7 @@ class PRSimilarIssue: level=IssueLevel.COMMENT) ) corpus.append(comment_record) - df = pd.DataFrame(corpus.dict()["documents"]) + df = pd.DataFrame(corpus.model_dump()["documents"]) get_logger().info('Done') get_logger().info('Embedding...') @@ -633,7 +633,7 @@ class PRSimilarIssue: ) corpus.append(comment_record) - df = pd.DataFrame(corpus.dict()["documents"]) + df = pd.DataFrame(corpus.model_dump()["documents"]) get_logger().info('Done') get_logger().info('Embedding...')