wbfdm 1.50.7__py2.py3-none-any.whl → 1.50.9__py2.py3-none-any.whl

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

{wbfdm-1.50.7.dist-info → wbfdm-1.50.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wbfdm
-Version: 1.50.7
+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.7.dist-info → wbfdm-1.50.9.dist-info}/RECORD

@@ -237,7 +237,7 @@ wbfdm/models/instruments/private_equities.py,sha256=uzwZi8IkmCKAHVTxnuFya9tehx7k
 wbfdm/models/instruments/querysets.py,sha256=c9fbHVvIRyaFU7jm_CM90lfIVzB8beOD8YII9-FqdT4,7191
 wbfdm/models/instruments/utils.py,sha256=88jnWINSSC0OwH-mCEOPLZXuhBCtEsxBpSaZ38GteaE,1365
 wbfdm/models/instruments/llm/__init__.py,sha256=dSmxRmEWb0A4O_lUoWuRKt2mBtUuLCTPVVJqGyi_n40,52
-wbfdm/models/instruments/llm/create_instrument_news_relationships.py,sha256=Eza39rlkNJxpURroIsJLImKC6F-KtTmkdjHn1kv4F3Q,3439
+wbfdm/models/instruments/llm/create_instrument_news_relationships.py,sha256=1EOd3VmLk31eIDzWLxKp9cRHg3c0IGHJkkwJTG_pxoU,3611
 wbfdm/models/instruments/mixin/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wbfdm/models/instruments/mixin/financials_computed.py,sha256=L5wjXDsR0maiwfOKP6KyWNJNH4nGOoAjSc_hDM7fsj0,35105
 wbfdm/models/instruments/mixin/financials_serializer_fields.py,sha256=-OkpcUt1rZmB3nUcO2vckpJdVm8IxRqkPDEgcPqqoRU,68886
@@ -352,6 +352,6 @@ wbfdm/viewsets/statements/__init__.py,sha256=odxtFYUDICPmz8WCE3nx93EvKZLSPBEI4d7
 wbfdm/viewsets/statements/statements.py,sha256=kmtM0uZ3f7eJJe5gVmd-iVra9dHwTB9x12p7f5qTEx8,4084
 wbfdm/viewsets/technical_analysis/__init__.py,sha256=qtCIBg0uSiZeJq_1tEQFilnorMBkMe6uCMfqar6-cLE,77
 wbfdm/viewsets/technical_analysis/monthly_performances.py,sha256=O1j8CGfOranL74LqVvcf7jERaDIboEJZiBf_AbbVDQ8,3974
-wbfdm-1.50.7.dist-info/METADATA,sha256=9UrsrlKQIHXGC3MCMbZgyw86lyMoQwcvApFlYre1olk,737
-wbfdm-1.50.7.dist-info/WHEEL,sha256=tkmg4JIqwd9H8mL30xA7crRmoStyCtGp0VWshokd1Jc,105
-wbfdm-1.50.7.dist-info/RECORD,,
+wbfdm-1.50.9.dist-info/METADATA,sha256=8QJvuseP4Skk5G-3N9GPPLyzPh8pI1awjM9W46HHGpA,737
+wbfdm-1.50.9.dist-info/WHEEL,sha256=tkmg4JIqwd9H8mL30xA7crRmoStyCtGp0VWshokd1Jc,105
+wbfdm-1.50.9.dist-info/RECORD,,