mirror of https://github.com/corca-ai/EVAL
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
22 lines
596 B
Python
22 lines
596 B
Python
import pandas as pd
|
|
|
|
from core.prompts.file import DATAFRAME_PROMPT
|
|
|
|
from .base import BaseHandler
|
|
|
|
|
|
class CsvToDataframe(BaseHandler):
|
|
def handle(self, filename: str):
|
|
df = pd.read_csv(filename)
|
|
description = (
|
|
f"Dataframe with {len(df)} rows and {len(df.columns)} columns. "
|
|
"Columns are: "
|
|
f"{', '.join(df.columns)}"
|
|
)
|
|
|
|
print(
|
|
f"\nProcessed CsvToDataframe, Input CSV: {filename}, Output Description: {description}"
|
|
)
|
|
|
|
return DATAFRAME_PROMPT.format(filename=filename, description=description)
|