Upload 14 files

resume_screener_pack/README.md

@@ -0,0 +1,2 @@
+DO NOT DELETE
+This readme file is needed to install from pyproject.toml.

resume_screener_pack/llama_index/packs/resume_screener/.gradio/certificate.pem

@@ -0,0 +1,31 @@
+-----BEGIN CERTIFICATE-----
+MIIFazCCA1OgAwIBAgIRAIIQz7DSQONZRGPgu2OCiwAwDQYJKoZIhvcNAQELBQAw
+TzELMAkGA1UEBhMCVVMxKTAnBgNVBAoTIEludGVybmV0IFNlY3VyaXR5IFJlc2Vh
+cmNoIEdyb3VwMRUwEwYDVQQDEwxJU1JHIFJvb3QgWDEwHhcNMTUwNjA0MTEwNDM4
+WhcNMzUwNjA0MTEwNDM4WjBPMQswCQYDVQQGEwJVUzEpMCcGA1UEChMgSW50ZXJu
+ZXQgU2VjdXJpdHkgUmVzZWFyY2ggR3JvdXAxFTATBgNVBAMTDElTUkcgUm9vdCBY
+MTCCAiIwDQYJKoZIhvcNAQEBBQADggIPADCCAgoCggIBAK3oJHP0FDfzm54rVygc
+h77ct984kIxuPOZXoHj3dcKi/vVqbvYATyjb3miGbESTtrFj/RQSa78f0uoxmyF+
+0TM8ukj13Xnfs7j/EvEhmkvBioZxaUpmZmyPfjxwv60pIgbz5MDmgK7iS4+3mX6U
+A5/TR5d8mUgjU+g4rk8Kb4Mu0UlXjIB0ttov0DiNewNwIRt18jA8+o+u3dpjq+sW
+T8KOEUt+zwvo/7V3LvSye0rgTBIlDHCNAymg4VMk7BPZ7hm/ELNKjD+Jo2FR3qyH
+B5T0Y3HsLuJvW5iB4YlcNHlsdu87kGJ55tukmi8mxdAQ4Q7e2RCOFvu396j3x+UC
+B5iPNgiV5+I3lg02dZ77DnKxHZu8A/lJBdiB3QW0KtZB6awBdpUKD9jf1b0SHzUv
+KBds0pjBqAlkd25HN7rOrFleaJ1/ctaJxQZBKT5ZPt0m9STJEadao0xAH0ahmbWn
+OlFuhjuefXKnEgV4We0+UXgVCwOPjdAvBbI+e0ocS3MFEvzG6uBQE3xDk3SzynTn
+jh8BCNAw1FtxNrQHusEwMFxIt4I7mKZ9YIqioymCzLq9gwQbooMDQaHWBfEbwrbw
+qHyGO0aoSCqI3Haadr8faqU9GY/rOPNk3sgrDQoo//fb4hVC1CLQJ13hef4Y53CI
+rU7m2Ys6xt0nUW7/vGT1M0NPAgMBAAGjQjBAMA4GA1UdDwEB/wQEAwIBBjAPBgNV
+HRMBAf8EBTADAQH/MB0GA1UdDgQWBBR5tFnme7bl5AFzgAiIyBpY9umbbjANBgkq
+hkiG9w0BAQsFAAOCAgEAVR9YqbyyqFDQDLHYGmkgJykIrGF1XIpu+ILlaS/V9lZL
+ubhzEFnTIZd+50xx+7LSYK05qAvqFyFWhfFQDlnrzuBZ6brJFe+GnY+EgPbk6ZGQ
+3BebYhtF8GaV0nxvwuo77x/Py9auJ/GpsMiu/X1+mvoiBOv/2X/qkSsisRcOj/KK
+NFtY2PwByVS5uCbMiogziUwthDyC3+6WVwW6LLv3xLfHTjuCvjHIInNzktHCgKQ5
+ORAzI4JMPJ+GslWYHb4phowim57iaztXOoJwTdwJx4nLCgdNbOhdjsnvzqvHu7Ur
+TkXWStAmzOVyyghqpZXjFaH3pO3JLF+l+/+sKAIuvtd7u+Nxe5AW0wdeRlN8NwdC
+jNPElpzVmbUq4JUagEiuTDkHzsxHpFKVK7q4+63SM1N95R1NbdWhscdCb+ZAJzVc
+oyi3B43njTOQ5yOf+1CceWxG1bQVs5ZufpsMljq4Ui0/1lvh+wjChP4kqKOJ2qxq
+4RgqsahDYVvTH9w7jXbyLeiNdd8XM2w9U/t7y0Ff/9yi0GE44Za4rF2LN9d11TPA
+mRGunUHBcnWEvgJBQl9nJEiU0Zsnvgc/ubhPgXRR4Xq37Z0j4r7g1SgEEzwxA57d
+emyPxgcYxn/eR44/KJ4EBs+lVDR3veyJm+kXQ99b21/+jh5Xos1AnX5iItreGCc=
+-----END CERTIFICATE-----

resume_screener_pack/llama_index/packs/resume_screener/BUILD

@@ -0,0 +1 @@
+python_sources()

resume_screener_pack/llama_index/packs/resume_screener/__init__.py

@@ -0,0 +1,3 @@
+from llama_index.packs.resume_screener.base import ResumeScreenerPack
+
+__all__ = ["ResumeScreenerPack"]

resume_screener_pack/llama_index/packs/resume_screener/base.py

@@ -0,0 +1,82 @@
+from pathlib import Path
+from typing import Any, Dict, List, Optional
+
+from llama_index.core import Settings
+from llama_index.core.llama_pack.base import BaseLlamaPack
+from llama_index.core.response_synthesizers import TreeSummarize
+from llama_index.core.schema import NodeWithScore
+from llama_index.llms.openai import OpenAI
+from llama_index.readers.file import PDFReader
+from llama_index.core.bridge.pydantic import BaseModel, Field
+
+# backwards compatibility
+try:
+    from llama_index.core.llms.llm import LLM
+except ImportError:
+    from llama_index.core.llms.base import LLM
+
+QUERY_TEMPLATE = """
+You are an expert resume reviewer.
+You job is to decide if the candidate pass the resume screen given the job description and a list of criteria:
+
+### Job Description
+{job_description}
+
+### Screening Criteria
+{criteria_str}
+"""
+
+
+class CriteriaDecision(BaseModel):
+    """The decision made based on a single criteria."""
+
+    decision: bool = Field(description="The decision made based on the criteria")
+    reasoning: str = Field(description="The reasoning behind the decision")
+
+
+class ResumeScreenerDecision(BaseModel):
+    """The decision made by the resume screener."""
+
+    criteria_decisions: List[CriteriaDecision] = Field(
+        description="The decisions made based on the criteria"
+    )
+    overall_reasoning: str = Field(
+        description="The reasoning behind the overall decision"
+    )
+    overall_decision: bool = Field(
+        description="The overall decision made based on the criteria"
+    )
+
+
+def _format_criteria_str(criteria: List[str]) -> str:
+    criteria_str = ""
+    for criterion in criteria:
+        criteria_str += f"- {criterion}\n"
+    return criteria_str
+
+
+class ResumeScreenerPack(BaseLlamaPack):
+    def __init__(
+        self, job_description: str, criteria: List[str], llm: Optional[LLM] = None
+    ) -> None:
+        self.reader = PDFReader()
+        llm = llm or OpenAI(model="gpt-3.5-turbo")
+        Settings.llm = llm
+        self.synthesizer = TreeSummarize(output_cls=ResumeScreenerDecision)
+        criteria_str = _format_criteria_str(criteria)
+        self.query = QUERY_TEMPLATE.format(
+            job_description=job_description, criteria_str=criteria_str
+        )
+
+    def get_modules(self) -> Dict[str, Any]:
+        """Get modules."""
+        return {"reader": self.reader, "synthesizer": self.synthesizer}
+
+    def run(self, resume_path: str, *args: Any, **kwargs: Any) -> Any:
+        """Run pack."""
+        docs = self.reader.load_data(Path(resume_path))
+        output = self.synthesizer.synthesize(
+            query=self.query,
+            nodes=[NodeWithScore(node=doc, score=1.0) for doc in docs],
+        )
+        return output.response

resume_screener_pack/llama_index/packs/resume_screener/resume_screener_pack/README.md

@@ -0,0 +1,2 @@
+DO NOT DELETE
+This readme file is needed to install from pyproject.toml.

resume_screener_pack/llama_index/packs/resume_screener/resume_screener_pack/llama_index/packs/resume_screener/BUILD

@@ -0,0 +1 @@
+python_sources()

resume_screener_pack/llama_index/packs/resume_screener/resume_screener_pack/llama_index/packs/resume_screener/__init__.py

@@ -0,0 +1,3 @@
+from llama_index.packs.resume_screener.base import ResumeScreenerPack
+
+__all__ = ["ResumeScreenerPack"]

resume_screener_pack/llama_index/packs/resume_screener/resume_screener_pack/llama_index/packs/resume_screener/base.py

@@ -0,0 +1,82 @@
+from pathlib import Path
+from typing import Any, Dict, List, Optional
+
+from llama_index.core import Settings
+from llama_index.core.llama_pack.base import BaseLlamaPack
+from llama_index.core.response_synthesizers import TreeSummarize
+from llama_index.core.schema import NodeWithScore
+from llama_index.llms.openai import OpenAI
+from llama_index.readers.file import PDFReader
+from llama_index.core.bridge.pydantic import BaseModel, Field
+
+# backwards compatibility
+try:
+    from llama_index.core.llms.llm import LLM
+except ImportError:
+    from llama_index.core.llms.base import LLM
+
+QUERY_TEMPLATE = """
+You are an expert resume reviewer.
+You job is to decide if the candidate pass the resume screen given the job description and a list of criteria:
+
+### Job Description
+{job_description}
+
+### Screening Criteria
+{criteria_str}
+"""
+
+
+class CriteriaDecision(BaseModel):
+    """The decision made based on a single criteria."""
+
+    decision: bool = Field(description="The decision made based on the criteria")
+    reasoning: str = Field(description="The reasoning behind the decision")
+
+
+class ResumeScreenerDecision(BaseModel):
+    """The decision made by the resume screener."""
+
+    criteria_decisions: List[CriteriaDecision] = Field(
+        description="The decisions made based on the criteria"
+    )
+    overall_reasoning: str = Field(
+        description="The reasoning behind the overall decision"
+    )
+    overall_decision: bool = Field(
+        description="The overall decision made based on the criteria"
+    )
+
+
+def _format_criteria_str(criteria: List[str]) -> str:
+    criteria_str = ""
+    for criterion in criteria:
+        criteria_str += f"- {criterion}\n"
+    return criteria_str
+
+
+class ResumeScreenerPack(BaseLlamaPack):
+    def __init__(
+        self, job_description: str, criteria: List[str], llm: Optional[LLM] = None
+    ) -> None:
+        self.reader = PDFReader()
+        llm = llm or OpenAI(model="gpt-4")
+        Settings.llm = llm
+        self.synthesizer = TreeSummarize(output_cls=ResumeScreenerDecision)
+        criteria_str = _format_criteria_str(criteria)
+        self.query = QUERY_TEMPLATE.format(
+            job_description=job_description, criteria_str=criteria_str
+        )
+
+    def get_modules(self) -> Dict[str, Any]:
+        """Get modules."""
+        return {"reader": self.reader, "synthesizer": self.synthesizer}
+
+    def run(self, resume_path: str, *args: Any, **kwargs: Any) -> Any:
+        """Run pack."""
+        docs = self.reader.load_data(Path(resume_path))
+        output = self.synthesizer.synthesize(
+            query=self.query,
+            nodes=[NodeWithScore(node=doc, score=1.0) for doc in docs],
+        )
+        return output.response

resume_screener_pack/llama_index/packs/resume_screener/resume_screener_pack/pyproject.toml

@@ -0,0 +1,66 @@
+[build-system]
+build-backend = "poetry.core.masonry.api"
+requires = ["poetry-core"]
+
+[tool.codespell]
+check-filenames = true
+check-hidden = true
+skip = "*.csv,*.html,*.json,*.jsonl,*.pdf,*.txt,*.ipynb"
+
+[tool.llamahub]
+contains_example = false
+import_path = "llama_index.packs.resume_screener"
+
+[tool.llamahub.class_authors]
+ResumeScreenerPack = "Disiok"
+
+[tool.mypy]
+disallow_untyped_defs = true
+exclude = ["_static", "build", "examples", "notebooks", "venv"]
+ignore_missing_imports = true
+python_version = "3.8"
+
+[tool.poetry]
+authors = ["Your Name <[email protected]>"]
+description = "llama-index packs resume_screener integration"
+exclude = ["**/BUILD"]
+keywords = ["document", "pdf", "resume", "structured output"]
+license = "MIT"
+maintainers = ["Disiok"]
+name = "llama-index-packs-resume-screener"
+readme = "README.md"
+version = "0.4.0"
+
+[tool.poetry.dependencies]
+python = ">=3.8.1,<4.0"
+pypdf = "^4.0.1"
+llama-index-readers-file = "^0.2.0"
+llama-index-llms-openai = "^0.2.0"
+
+[tool.poetry.group.dev.dependencies]
+ipython = "8.10.0"
+jupyter = "^1.0.0"
+mypy = "0.991"
+pre-commit = "3.2.0"
+pylint = "2.15.10"
+pytest = "7.2.1"
+pytest-mock = "3.11.1"
+ruff = "0.0.292"
+tree-sitter-languages = "^1.8.0"
+types-Deprecated = ">=0.1.0"
+types-PyYAML = "^6.0.12.12"
+types-protobuf = "^4.24.0.4"
+types-redis = "4.5.5.0"
+types-requests = "2.28.11.8"
+types-setuptools = "67.1.0.0"
+
+[tool.poetry.group.dev.dependencies.black]
+extras = ["jupyter"]
+version = "<=23.9.1,>=23.7.0"
+
+[tool.poetry.group.dev.dependencies.codespell]
+extras = ["toml"]
+version = ">=v2.2.6"
+
+[[tool.poetry.packages]]
+include = "llama_index/"

resume_screener_pack/pyproject.toml

@@ -0,0 +1,66 @@
+[build-system]
+build-backend = "poetry.core.masonry.api"
+requires = ["poetry-core"]
+
+[tool.codespell]
+check-filenames = true
+check-hidden = true
+skip = "*.csv,*.html,*.json,*.jsonl,*.pdf,*.txt,*.ipynb"
+
+[tool.llamahub]
+contains_example = false
+import_path = "llama_index.packs.resume_screener"
+
+[tool.llamahub.class_authors]
+ResumeScreenerPack = "Disiok"
+
+[tool.mypy]
+disallow_untyped_defs = true
+exclude = ["_static", "build", "examples", "notebooks", "venv"]
+ignore_missing_imports = true
+python_version = "3.8"
+
+[tool.poetry]
+authors = ["Your Name <[email protected]>"]
+description = "llama-index packs resume_screener integration"
+exclude = ["**/BUILD"]
+keywords = ["document", "pdf", "resume", "structured output"]
+license = "MIT"
+maintainers = ["Disiok"]
+name = "llama-index-packs-resume-screener"
+readme = "README.md"
+version = "0.4.0"
+
+[tool.poetry.dependencies]
+python = ">=3.8.1,<4.0"
+pypdf = "^4.0.1"
+llama-index-readers-file = "^0.2.0"
+llama-index-llms-openai = "^0.2.0"
+
+[tool.poetry.group.dev.dependencies]
+ipython = "8.10.0"
+jupyter = "^1.0.0"
+mypy = "0.991"
+pre-commit = "3.2.0"
+pylint = "2.15.10"
+pytest = "7.2.1"
+pytest-mock = "3.11.1"
+ruff = "0.0.292"
+tree-sitter-languages = "^1.8.0"
+types-Deprecated = ">=0.1.0"
+types-PyYAML = "^6.0.12.12"
+types-protobuf = "^4.24.0.4"
+types-redis = "4.5.5.0"
+types-requests = "2.28.11.8"
+types-setuptools = "67.1.0.0"
+
+[tool.poetry.group.dev.dependencies.black]
+extras = ["jupyter"]
+version = "<=23.9.1,>=23.7.0"
+
+[tool.poetry.group.dev.dependencies.codespell]
+extras = ["toml"]
+version = ">=v2.2.6"
+
+[[tool.poetry.packages]]
+include = "llama_index/"