jiandong's picture
Upload with huggingface_hub
dac7fdc
raw
history blame contribute delete
998 Bytes
import os
import openai
from typing import Dict, List, Optional
from pydantic import BaseModel, Field, SecretStr
from utils.summarizer import get_analyze_result
from workcell.integrations.types import MarkdownMixin
class Input(BaseModel):
keywords: List[str] = Field(default=["backend development", "micro services", 'dev-ops', 'OKR'], max_items=20,
description="List of keyword to describe your work this week.")
openai_api_key: Optional[SecretStr] = Field(
..., description="OpenAI API key, provide if you have, or use our default key (may reach tier limit)."
)
def generate_weekly_report(input: Input) -> MarkdownMixin:
"""Returns a weekly report by several keywords, generated by OpenAI GPT3 API."""
openai.api_key = os.getenv('SECRET_OPENAI_WORKCELL_WEBPAGE_QA')
# return summarization
text = input.keywords
analyze = get_analyze_result(text)
output = MarkdownMixin(
data = analyze
)
return output