Skip to content

Commit

Permalink
refactor: Rename EntityExtractor class to TextEntityExtractor
Browse files Browse the repository at this point in the history
  • Loading branch information
Appointat committed Oct 9, 2024
1 parent cfa424f commit 212077f
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions dbgpt/rag/transformer/entity_extractor.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
"""EntityExtractor class."""
"""TextEntityExtractor class."""

import logging
import re
Expand All @@ -10,15 +10,15 @@
logger = logging.getLogger(__name__)


class EntityExtractor(LLMExtractor):
class TextEntityExtractor(LLMExtractor):
"""
EntityExtractor class for extracting entities and relationships (with the provieded domain knowledge) from text.
TextEntityExtractor class for extracting entities and relationships (with the provieded domain knowledge) from text.
Inherits from LLMExtractor.
"""

def __init__(self, llm_client: LLMClient, model_name: str, domain_knowledge_pt: str):
"""
Initialize the EntityExtractor.
Initialize the TextEntityExtractor.
Args:
llm_client (LLMClient): The language model client.
Expand Down

0 comments on commit 212077f

Please sign in to comment.