Compare commits

..

2 Commits

Author SHA1 Message Date
Thamognya Kodi fe61b63a0b updae 2023-01-19 19:36:13 +07:00
Thamognya Kodi 5939e7804e
working 2023-01-19 14:47:35 +07:00
3 changed files with 7 additions and 8 deletions

View File

@ -47,7 +47,7 @@ def answer(
results: tuple[list[str], list[str]] = internet.Google( results: tuple[list[str], list[str]] = internet.Google(
query, GOOGLE_SEARCH_API_KEY, GOOGLE_SEARCH_ENGINE_ID query, GOOGLE_SEARCH_API_KEY, GOOGLE_SEARCH_ENGINE_ID
).google() ).google()
context: str = str(" ".join([str(string) for string in results[0]])) context: str = str(" ".join([str(string) for string in results]))
print(f"context: {context}") print(f"context: {context}")
if model.startswith("openai-"): if model.startswith("openai-"):
@ -90,6 +90,6 @@ def answer(
print( print(
answer( answer(
query="What is the newest pokemon game?", query="What is the newest pokemon game?",
model="hf-deepset/xlm-roberta-large-squad2", model="openai-chatgpt",
) )
) )

View File

@ -100,9 +100,7 @@ class Google:
soup = BeautifulSoup(html, "html.parser") soup = BeautifulSoup(html, "html.parser")
text = soup.get_text() text = soup.get_text()
normalized_text = normalizer(text) normalized_text = normalizer(text)
sentences: list[str] = sentencizer(normalized_text) return normalized_text
sentence: str = str(" ".join(sentences))
return sentence
except Exception: except Exception:
error: str = "" error: str = ""
return error return error
@ -118,8 +116,9 @@ class Google:
def __get_urls_contents(self: "Google") -> None: def __get_urls_contents(self: "Google") -> None:
loop = asyncio.new_event_loop() loop = asyncio.new_event_loop()
asyncio.set_event_loop(loop) asyncio.set_event_loop(loop)
self.__content = loop.run_until_complete(self.__fetch_urls(self.__urls)) contents = loop.run_until_complete(self.__fetch_urls(self.__urls))
loop.close() loop.close()
self.__content = contents
def __filter_irrelevant_processing(self: "Google") -> None: def __filter_irrelevant_processing(self: "Google") -> None:
with concurrent.futures.ThreadPoolExecutor(max_workers=500) as executor: with concurrent.futures.ThreadPoolExecutor(max_workers=500) as executor:
@ -137,7 +136,7 @@ class Google:
self.__get_urls_contents() self.__get_urls_contents()
if filter_irrelevant: if filter_irrelevant:
self.__filter_irrelevant_processing() self.__filter_irrelevant_processing()
results: tuple[list[str], list[str]] = (self.__content[0], self.__urls) # type: ignore results: tuple[list[str], list[str]] = (self.__content, self.__urls) # type: ignore
return results return results

View File

@ -5,7 +5,7 @@ build-backend = "poetry.core.masonry.api"
[tool.poetry] [tool.poetry]
name = "internet-ml" name = "internet-ml"
version = "0.1.0" version = "3.0.0" # fresh start from 3.0.0
description = "Internet-ML: Allowing ML to connect to the internet" description = "Internet-ML: Allowing ML to connect to the internet"
readme = "./.github/README.md" readme = "./.github/README.md"
authors = ["Thamognya Kodi <contact@thamognya.com>"] authors = ["Thamognya Kodi <contact@thamognya.com>"]