3fiKTdZddlmZmZmZmZddlmZddlm Z m Z Gdde e Z y)0Simple in memory docstore in the form of a dict.)DictListOptionalUnion)Document) AddableMixinDocstorecreZdZdZd deeeeffdZdeeefddfdZ de ddfd Z d ede eeffd Z y) InMemoryDocstorerN_dictc&|||_yi|_y)zInitialize with dict.Nr )selfr s d/var/www/auto_recruiter/arenv/lib/python3.12/site-packages/langchain_community/docstore/in_memory.py__init__zInMemoryDocstore.__init__ s#/U R textsreturnct|j|j}|rtd|i|j||_y)zAdd texts to in memory dictionary. Args: texts: dictionary of id -> document. Returns: None z%Tried to add ids that already exist: N)set intersectionr ValueError)rr overlappings raddzInMemoryDocstore.addsG%j--djj9 D[MRS S, ,e, ridsct|j|j}|std||D]}|jj |y)z'Deleting IDs from in memory dictionary.z*Tried to delete ids that does not exist: N)rrr rpop)rrr_ids rdeletezInMemoryDocstore.deletesO#h++DJJ7 I#OP P C JJNN3  rsearchcH||jvrd|dS|j|S)zSearch via direct lookup. Args: search: id of a document to search for. Returns: Document if found, else error message. zID z not found.r)rr!s rr!zInMemoryDocstore.search's.  # , ,::f% %r)N)__name__ __module__ __qualname____doc__rrstrrrrrr rr!rrr r sj:8htCM':;8 -c8m, - - $ 4  &S &U3=%9 &rr N) r&typingrrrrlangchain_core.documentsr!langchain_community.docstore.baser r r r(rrr,s"6..-D)&x)&r