LLM based classification
Classification module
Large Language Model based classification main handlers.
embedding.EmbeddingHandler | Handles embedding operations for the Chroma vector store. |
llm.ClassificationLLM | Wraps the logic for using an LLM to classify respondent’s data |
prompt.PromptTemplates | Class to store prompt templates for SOC and SIC classification tasks. |
Response models
SocCandidate | Represents a candidate SOC code based on provided job title and description. |
SocResponse | Represents a response model for SOC code assignment. |
SicCandidate | Represents a candidate SIC code with associated information. |
SicResponse | Represents a response model for SIC code assignment. |
RagCandidate | Represents a candidate classification code with associated information. |
RagResponse | Represents a response model for classification code assignment. |
SIC Index Abstraction
Data models to represent Standard Industry Classiffication
SIC | Main class for SIC lookups. |
SicCode | Standard Industrial Classification code. |
SicNode | Tree data structure where the nodes hold all data associated with a given SIC. |
SIC metadata
sic_meta_model.ClassificationMeta | Represents a classification meta model. |
sicDB.SicMeta | SIC Meta data model class for SIC codes and their descriptions |
Helpers
Config and Log utils
setup_logging | Set up console and file logging. |
get_config | Fetch the configuration. |
check_file_exists | Check if the file exists. |