diff --git a/brancheneinstufung.py b/brancheneinstufung.py index afe69c69..7706d7a1 100644 --- a/brancheneinstufung.py +++ b/brancheneinstufung.py @@ -169,8 +169,11 @@ def get_wikipedia_data(name, website_hint=""): def classify_company(row, wikipedia_url=""): user_prompt = { "role": "user", - "content": f"{row[0]};{row[1]};{row[2]};{row[4]};{row[5]} -Wikipedia-Link: {wikipedia_url}" + "content": ( + f"{row[0]};{row[1]};{row[2]};{row[4]};{row[5]} +" + f"Wikipedia-Link: {wikipedia_url}" + ) } response = openai.chat.completions.create( model="gpt-3.5-turbo",