wbfdm 1.50.8__tar.gz → 1.50.9__tar.gz

{wbfdm-1.50.8 → wbfdm-1.50.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wbfdm
-Version: 1.50.8
+Version: 1.50.9
 Summary: The workbench module ensures rapid access to diverse financial data (market, fundamental, forecasts, ESG), with features for storing instruments, classifying them, and conducting financial analysis.
 Author-email: Christopher Wittlinger <c.wittlinger@stainly.com>
 Requires-Dist: roman==4.*

{wbfdm-1.50.8 → wbfdm-1.50.9}/wbfdm/models/instruments/llm/create_instrument_news_relationships.py

@@ -57,29 +57,31 @@ def run_company_extraction_llm(title: str, description: str, *args) -> list[dict
                 SystemMessage(
                     content="You will be parsed a news article, please provide the name of the publicly listed companies mentioned in the article, along with their ISIN, ticker, RIC, sentiment, and analysis."
                 ),
-                HumanMessage(content=f"Title: {title}, Description: {description}"),
+                HumanMessage(content="Title: {title}, Description: {description}"),
             ],
             output_model=CompaniesModel,
+            query={"title": title, "description": description},
         )
-        instrument_ct = ContentType.objects.get_for_model(Instrument)
-        for company in res.get("companies", []):
-            instrument = InstrumentLookup(Instrument).lookup(
-                only_security=True,
-                name=company.name,
-                isin=company.isin,
-                ticker=company.ticker,
-                refinitiv_identifier_code=company.refinitiv_identifier_code,
-            )
-            if instrument is not None:
-                relationships.append(
-                    {
-                        "content_type_id": instrument_ct.id,
-                        "object_id": instrument.get_root().id,
-                        "sentiment": company.sentiment,
-                        "analysis": company.analysis,
-                        "content_object_repr": str(instrument),
-                    }
+        if isinstance(res, CompaniesModel):
+            instrument_ct = ContentType.objects.get_for_model(Instrument)
+            for company in res.companies:
+                instrument = InstrumentLookup(Instrument).lookup(
+                    only_security=True,
+                    name=company.name,
+                    isin=company.isin,
+                    ticker=company.ticker,
+                    refinitiv_identifier_code=company.refinitiv_identifier_code,
                 )
+                if instrument is not None:
+                    relationships.append(
+                        {
+                            "content_type_id": instrument_ct.id,
+                            "object_id": instrument.get_root().id,
+                            "sentiment": company.sentiment,
+                            "analysis": company.analysis,
+                            "content_object_repr": str(instrument),
+                        }
+                    )
     except tuple(APIStatusErrors) as e:  # for APIStatusError, we let celery retry it
         raise e
     except Exception as e:  # otherwise we log the error and silently fail