Spaces:
Sleeping
Sleeping
Update TestFindingAgent.py
Browse files- TestFindingAgent.py +13 -0
TestFindingAgent.py
CHANGED
|
@@ -1,3 +1,16 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
class TestFindingAgent:
|
| 2 |
def __init__(self, hcc_code: str, model_version: str,
|
| 3 |
model: str = "gpt-4o", output_file: Optional[str] = None):
|
|
|
|
| 1 |
+
import os
|
| 2 |
+
import json
|
| 3 |
+
import pandas as pd
|
| 4 |
+
from PyPDF2 import PdfReader
|
| 5 |
+
from json_repair import repair_json
|
| 6 |
+
from typing import List, Dict, Any
|
| 7 |
+
from typing import List, Dict, Any, Optional
|
| 8 |
+
from crewai import Agent, Task, Crew, Process
|
| 9 |
+
from crewai_tools import SerperDevTool
|
| 10 |
+
from langchain_openai import ChatOpenAI, OpenAIEmbeddings
|
| 11 |
+
from langchain.text_splitter import RecursiveCharacterTextSplitter
|
| 12 |
+
from langchain_community.vectorstores import Chroma
|
| 13 |
+
|
| 14 |
class TestFindingAgent:
|
| 15 |
def __init__(self, hcc_code: str, model_version: str,
|
| 16 |
model: str = "gpt-4o", output_file: Optional[str] = None):
|